apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

fix import

This commit is contained in:
Sean Sube 2023-08-06 10:07:39 -05:00
parent a201a74b40
commit b1a554e1a7
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
api/onnx_web/diffusers/load.py
View File

@ -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