fix import
This commit is contained in:
parent
f2d0c2f765
commit
956a260db6
|
@ -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
|
||||||
|
|
||||||
|
|
Loading…
Reference in New Issue