fix

huggingface · Jul 18, 2023 · 420983d · 420983d
1 parent e455428
commit 420983d
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/optimum/onnxruntime/modeling_diffusion.py b/optimum/onnxruntime/modeling_diffusion.py
@@ -28,7 +28,7 @@
     LMSDiscreteScheduler,
     PNDMScheduler,
     StableDiffusionPipeline,
-    StableDiffusionXLPipeline,
+    StableDiffusionXLImg2ImgPipeline,
 )
 from diffusers.schedulers.scheduling_utils import SCHEDULER_CONFIG_NAME
 from diffusers.utils import CONFIG_NAME
@@ -547,7 +547,7 @@ def __call__(self, *args, **kwargs):
 
 
 class ORTStableDiffusionXLPipelineBase(ORTStableDiffusionPipelineBase):
-    auto_model_class = StableDiffusionXLPipeline
+    auto_model_class = StableDiffusionXLImg2ImgPipeline
 
     def __init__(
         self,