fix import
This commit is contained in:
parent
a201a74b40
commit
b1a554e1a7
|
@ -3,7 +3,10 @@ from os import path
|
|||
from typing import Any, List, Optional, Tuple
|
||||
|
||||
from onnx import load_model
|
||||
from optimum import ORTStableDiffusionXLPipeline, ORTStableDiffusionXLImg2ImgPipeline
|
||||
from optimum.onnxruntime import (
|
||||
ORTStableDiffusionXLImg2ImgPipeline,
|
||||
ORTStableDiffusionXLPipeline,
|
||||
)
|
||||
from transformers import CLIPTokenizer
|
||||
|
||||
from ..constants import ONNX_MODEL
|
||||
|
|
Loading…
Reference in New Issue