dezzman commited on
Commit
98378b3
·
verified ·
1 Parent(s): 40e2fde

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -26,8 +26,6 @@ def get_lora_sd_pipeline(
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
- pipe.unet.load_lora_weights(os.path.join(unet_sub_dir, "adapter_model.safetensors"))
30
- pipe.fuse_lora(lora_scale=0.4)
31
  after_params = pipe.unet.parameters()
32
  print("Parameters changed:", any(torch.any(b != a) for b, a in zip(before_params, after_params)))
33
 
@@ -75,6 +73,7 @@ def infer(
75
  lora_scale=0.5,
76
  progress=gr.Progress(track_tqdm=True)
77
  ):
 
78
  generator = torch.Generator(device).manual_seed(seed)
79
 
80
  if model_id != model_id_default:
 
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
 
 
73
  lora_scale=0.5,
74
  progress=gr.Progress(track_tqdm=True)
75
  ):
76
+
77
  generator = torch.Generator(device).manual_seed(seed)
78
 
79
  if model_id != model_id_default: