Spaces:

dezzman
/

diffusion_models

Running

dezzman commited on Feb 2

Commit

9b901a7

verified ·

1 Parent(s): de6e742

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -24,7 +24,7 @@ def get_lora_sd_pipeline(
     pipe = StableDiffusionPipeline.from_pretrained(base_model_name_or_path, torch_dtype=dtype)
     before_params = pipe.unet.parameters()
     pipe.unet = PeftModel.from_pretrained(pipe.unet, unet_sub_dir, adapter_name=adapter_name)
-    pipe_default.fuse_lora(lora_scale=0.5)
     after_params = pipe.unet.parameters()
     print("Parameters changed:", any(torch.any(b != a) for b, a in zip(before_params, after_params)))

     pipe = StableDiffusionPipeline.from_pretrained(base_model_name_or_path, torch_dtype=dtype)
     before_params = pipe.unet.parameters()
     pipe.unet = PeftModel.from_pretrained(pipe.unet, unet_sub_dir, adapter_name=adapter_name)
+    pipe.fuse_lora(lora_scale=0.5)
     after_params = pipe.unet.parameters()
     print("Parameters changed:", any(torch.any(b != a) for b, a in zip(before_params, after_params)))