diff --git a/pipeline/pipeline_controlnet_union_sd_xl.py b/pipeline/pipeline_controlnet_union_sd_xl.py index 9d9a125..c4ab4f8 100644 --- a/pipeline/pipeline_controlnet_union_sd_xl.py +++ b/pipeline/pipeline_controlnet_union_sd_xl.py @@ -28,7 +28,7 @@ from diffusers.image_processor import PipelineImageInput, VaeImageProcessor from diffusers.loaders import FromSingleFileMixin, StableDiffusionXLLoraLoaderMixin, TextualInversionLoaderMixin,IPAdapterMixin from diffusers.models import AutoencoderKL, ControlNetModel, UNet2DConditionModel,ImageProjection -from .controlnet_union import ControlNetModel_Union +from models.controlnet_union import ControlNetModel_Union from diffusers.models.attention_processor import ( AttnProcessor2_0, LoRAAttnProcessor2_0, @@ -1391,4 +1391,4 @@ def pack_weights(layers, prefix): # Copied from diffusers.pipelines.stable_diffusion_xl.pipeline_stable_diffusion_xl.StableDiffusionXLPipeline._remove_text_encoder_monkey_patch def _remove_text_encoder_monkey_patch(self): self._remove_text_encoder_monkey_patch_classmethod(self.text_encoder) - self._remove_text_encoder_monkey_patch_classmethod(self.text_encoder_2) \ No newline at end of file + self._remove_text_encoder_monkey_patch_classmethod(self.text_encoder_2)