Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -22,7 +22,10 @@ def get_lora_sd_pipeline(
|
|
22 |
raise ValueError("Please specify the base model name or path")
|
23 |
|
24 |
pipe = StableDiffusionPipeline.from_pretrained(base_model_name_or_path, torch_dtype=dtype)
|
|
|
25 |
pipe.unet = PeftModel.from_pretrained(pipe.unet, unet_sub_dir, adapter_name=adapter_name)
|
|
|
|
|
26 |
|
27 |
if os.path.exists(text_encoder_sub_dir):
|
28 |
pipe.text_encoder = PeftModel.from_pretrained(pipe.text_encoder, text_encoder_sub_dir, adapter_name=adapter_name)
|
|
|
22 |
raise ValueError("Please specify the base model name or path")
|
23 |
|
24 |
pipe = StableDiffusionPipeline.from_pretrained(base_model_name_or_path, torch_dtype=dtype)
|
25 |
+
before_params = pipe.unet.parameters()
|
26 |
pipe.unet = PeftModel.from_pretrained(pipe.unet, unet_sub_dir, adapter_name=adapter_name)
|
27 |
+
after_params = pipe.unet.parameters()
|
28 |
+
print("Parameters changed:", any(torch.any(b != a) for b, a in zip(before_params, after_params)))
|
29 |
|
30 |
if os.path.exists(text_encoder_sub_dir):
|
31 |
pipe.text_encoder = PeftModel.from_pretrained(pipe.text_encoder, text_encoder_sub_dir, adapter_name=adapter_name)
|