Update app.py

app.py

@@ -26,6 +26,8 @@ def get_lora_sd_pipeline(
     before_params = pipe.unet.parameters()
     pipe.unet = PeftModel.from_pretrained(pipe.unet, unet_sub_dir, adapter_name=adapter_name)
     pipe.unet.set_adapter(adapter_name)
+    pipe.load_lora_weights(os.path.join(unet_sub_dir, "adapter_model.safetensors"))
+    pipe.fuse_lora(lora_scale=0.4)
     after_params = pipe.unet.parameters()
     print("Parameters changed:", any(torch.any(b != a) for b, a in zip(before_params, after_params)))
     
@@ -87,8 +89,7 @@ def infer(
         prompt_embeds, negative_prompt_embeds = align_embeddings(prompt_embeds, negative_prompt_embeds)
         print(f"LoRA adapter loaded: {pipe.unet.active_adapters}")
         print(f"LoRA scale applied: {lora_scale}")
-        # pipe.fuse_lora(lora_scale=lora_scale)
-        pipe.unet = pipe.unet.merge_and_unload(lora_scale=lora_scale)
+        pipe.fuse_lora(lora_scale=lora_scale)
     
     params = {
         'prompt_embeds': prompt_embeds,