apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

fix import

This commit is contained in:
Sean Sube 2023-09-10 12:15:39 -05:00
parent f2d0c2f765
commit 956a260db6
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
1 changed files with 1 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
api/onnx_web/convert/diffusion/diffusion_xl.py
View File

@ -3,10 +3,8 @@ from os import path
from typing import Dict, Optional, Tuple from typing import Dict, Optional, Tuple
import torch import torch
from diffusers import StableDiffusionXLPipeline
from optimum.exporters.onnx import main_export from optimum.exporters.onnx import main_export
from optimum.pipelines.diffusers.pipeline_stable_diffusion_xl import (
StableDiffusionXLPipeline,
)
from ..utils import ConversionContext from ..utils import ConversionContext