Real-Time-LCM-ControlNet-Lora-SD1.5

Runtime error

radames commited on Dec 2, 2023

Commit

d789056

1 Parent(s): 2398f39

fix

Files changed (1) hide show

pipelines/controlnetLoraSD15.py CHANGED Viewed

@@ -180,7 +180,7 @@ class Pipeline:
                     base_model_id,
                     controlnet=controlnet_canny,
                 )
-            self.pipes[base_model_id] = pipe
         else:
             for base_model_id in base_models.keys():
                 pipe = StableDiffusionControlNetImg2ImgPipeline.from_pretrained(

                     base_model_id,
                     controlnet=controlnet_canny,
                 )
+                self.pipes[base_model_id] = pipe
         else:
             for base_model_id in base_models.keys():
                 pipe = StableDiffusionControlNetImg2ImgPipeline.from_pretrained(