Spaces:

YaArtemNosenko
/

diffusion

Sleeping

YaArtemNosenko commited on Feb 28

Commit

e8eaea5

verified ·

1 Parent(s): 34f7ee5

[RENAME] Rename stablediffusion

Files changed (1) hide show

app.py CHANGED Viewed

@@ -102,7 +102,7 @@ def load_pipeline(model_id,
         pipe.unet.load_state_dict({k: lora_scale * v for k, v in pipe.unet.state_dict().items()})
         pipe.text_encoder.load_state_dict({k: lora_scale * v for k, v in pipe.text_encoder.state_dict().items()})
     else:
-        pipe = DiffusionPipeline.from_pretrained(model_id,
                                                  torch_dtype=torch_dtype
                                                  )

         pipe.unet.load_state_dict({k: lora_scale * v for k, v in pipe.unet.state_dict().items()})
         pipe.text_encoder.load_state_dict({k: lora_scale * v for k, v in pipe.text_encoder.state_dict().items()})
     else:
+        pipe = StableDiffusionPipeline.from_pretrained(model_id,
                                                  torch_dtype=torch_dtype
                                                  )