MrAlex
/

ControlNetImg2ImgPipeline

Model card Files Files and versions

MrAlex commited on Apr 21, 2023

Commit

f413d0c

·

1 Parent(s): 38fcd44

Fix imports

Files changed (1) hide show

pipeline.py +2 -1

pipeline.py CHANGED Viewed

@@ -8,7 +8,8 @@ import PIL.Image
 import torch
 from transformers import CLIPImageProcessor, CLIPTextModel, CLIPTokenizer
-from diffusers.loaders import FromCkptMixin, LoraLoaderMixin, TextualInversionLoaderMixin
 from diffusers import AutoencoderKL, ControlNetModel, DiffusionPipeline, UNet2DConditionModel, logging
 from diffusers.pipelines.stable_diffusion import StableDiffusionPipelineOutput, StableDiffusionSafetyChecker

 import torch
 from transformers import CLIPImageProcessor, CLIPTextModel, CLIPTokenizer
+from diffusers.loaders import TextualInversionLoaderMixin
+# from diffusers.loaders import FromCkptMixin, LoraLoaderMixin, TextualInversionLoaderMixin
 from diffusers import AutoencoderKL, ControlNetModel, DiffusionPipeline, UNet2DConditionModel, logging
 from diffusers.pipelines.stable_diffusion import StableDiffusionPipelineOutput, StableDiffusionSafetyChecker