primerz commited on
Commit
b034233
·
verified ·
1 Parent(s): cc0ae1f

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
- from diffusers.pipelines.controlnet.multicontrolnet import MultiControlNetModel
 
 
 
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)