Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -25,8 +25,7 @@ def get_lora_sd_pipeline(
|
|
| 25 |
pipe = StableDiffusionPipeline.from_pretrained(base_model_name_or_path, torch_dtype=dtype)
|
| 26 |
before_params = pipe.unet.parameters()
|
| 27 |
pipe.unet = PeftModel.from_pretrained(pipe.unet, unet_sub_dir, adapter_name=adapter_name)
|
| 28 |
-
pipe.
|
| 29 |
-
pipe.unet.set_adapter(adapter_name, lora_scale=0.5)
|
| 30 |
after_params = pipe.unet.parameters()
|
| 31 |
print("Parameters changed:", any(torch.any(b != a) for b, a in zip(before_params, after_params)))
|
| 32 |
|
|
|
|
| 25 |
pipe = StableDiffusionPipeline.from_pretrained(base_model_name_or_path, torch_dtype=dtype)
|
| 26 |
before_params = pipe.unet.parameters()
|
| 27 |
pipe.unet = PeftModel.from_pretrained(pipe.unet, unet_sub_dir, adapter_name=adapter_name)
|
| 28 |
+
pipe.unet.set_adapter(adapter_name)
|
|
|
|
| 29 |
after_params = pipe.unet.parameters()
|
| 30 |
print("Parameters changed:", any(torch.any(b != a) for b, a in zip(before_params, after_params)))
|
| 31 |
|