Spaces:
Running
on
Zero
Running
on
Zero
Update pipeline_stable_diffusion_xl_instantid_img2img.py
Browse files
pipeline_stable_diffusion_xl_instantid_img2img.py
CHANGED
|
@@ -25,7 +25,10 @@ import torch.nn as nn
|
|
| 25 |
from diffusers import StableDiffusionXLControlNetImg2ImgPipeline
|
| 26 |
from diffusers.image_processor import PipelineImageInput
|
| 27 |
from diffusers.models import ControlNetModel
|
| 28 |
-
|
|
|
|
|
|
|
|
|
|
| 29 |
from diffusers.pipelines.stable_diffusion_xl import StableDiffusionXLPipelineOutput
|
| 30 |
from diffusers.utils import (
|
| 31 |
deprecate,
|
|
@@ -1069,4 +1072,4 @@ class StableDiffusionXLInstantIDImg2ImgPipeline(StableDiffusionXLControlNetImg2I
|
|
| 1069 |
if not return_dict:
|
| 1070 |
return (image,)
|
| 1071 |
|
| 1072 |
-
return StableDiffusionXLPipelineOutput(images=image)
|
|
|
|
| 25 |
from diffusers import StableDiffusionXLControlNetImg2ImgPipeline
|
| 26 |
from diffusers.image_processor import PipelineImageInput
|
| 27 |
from diffusers.models import ControlNetModel
|
| 28 |
+
# --- START FIX ---
|
| 29 |
+
# Use the new, correct import path to match model.py
|
| 30 |
+
from diffusers.models.controlnets.multicontrolnet import MultiControlNetModel
|
| 31 |
+
# --- END FIX ---
|
| 32 |
from diffusers.pipelines.stable_diffusion_xl import StableDiffusionXLPipelineOutput
|
| 33 |
from diffusers.utils import (
|
| 34 |
deprecate,
|
|
|
|
| 1072 |
if not return_dict:
|
| 1073 |
return (image,)
|
| 1074 |
|
| 1075 |
+
return StableDiffusionXLPipelineOutput(images=image)
|