Update app.py

app.py

@@ -108,7 +108,9 @@ def run_lora(prompt, cfg_scale, steps, selected_index, randomize_seed, seed, wid
             pipe.load_lora_weights(lora_path, weight_name=selected_lora["weights"])
         else:
             pipe.load_lora_weights(lora_path)
-
+    with calculateDuration(f"Fusing LoRA weights for {selected_lora['title']}"):
+        pipe.fuse_lora(lora_scale)
+    
     # Set random seed for reproducibility
     with calculateDuration("Randomizing seed"):
         if randomize_seed:
@@ -122,8 +124,9 @@ def run_lora(prompt, cfg_scale, steps, selected_index, randomize_seed, seed, wid
         final_image = image
         yield image, seed  # Yield intermediate images and seed
 
-    pipe.to("cpu")
-    pipe.unload_lora_weights()
+    with calculateDuration("Unfuse and unload"):
+        pipe.unfuse_lora()
+        pipe.unload_lora_weights()
     
     return final_image, seed  # Return the final image and seed