dezzman commited on
Commit
a677b1e
·
verified ·
1 Parent(s): b5602a4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -0
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)