Update app.py

app.py

@@ -70,7 +70,8 @@ def infer(
     model_id='CompVis/stable-diffusion-v1-4', 
     seed=42, 
     guidance_scale=7.0, 
-    lora_scale=0.5
+    lora_scale=0.5,
+    progress=gr.Progress(track_tqdm=True)
     ):
     generator = torch.Generator(device).manual_seed(seed)
     
@@ -86,7 +87,8 @@ 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.fuse_lora(lora_scale=lora_scale)
+        pipe.unet = pipe.unet.merge_and_unload(lora_scale=lora_scale)
     
     params = {
         'prompt_embeds': prompt_embeds,