Spaces:
Running
on
Zero
Running
on
Zero
Commit
•
03ac1d3
1
Parent(s):
751f5a5
Update app.py
Browse files
app.py
CHANGED
@@ -109,8 +109,8 @@ def run_lora(prompt, cfg_scale, steps, selected_index, randomize_seed, seed, wid
|
|
109 |
else:
|
110 |
pipe.load_lora_weights(lora_path)
|
111 |
|
112 |
-
with calculateDuration(f"Fusing LoRA weights for {selected_lora['title']}"):
|
113 |
-
pipe.fuse_lora(lora_scale)
|
114 |
|
115 |
# Set random seed for reproducibility
|
116 |
with calculateDuration("Randomizing seed"):
|
@@ -125,8 +125,8 @@ def run_lora(prompt, cfg_scale, steps, selected_index, randomize_seed, seed, wid
|
|
125 |
final_image = image
|
126 |
yield image, seed # Yield intermediate images and seed
|
127 |
|
128 |
-
with calculateDuration("
|
129 |
-
pipe.unfuse_lora()
|
130 |
pipe.unload_lora_weights()
|
131 |
|
132 |
return final_image, seed # Return the final image and seed
|
|
|
109 |
else:
|
110 |
pipe.load_lora_weights(lora_path)
|
111 |
|
112 |
+
#with calculateDuration(f"Fusing LoRA weights for {selected_lora['title']}"):
|
113 |
+
#pipe.fuse_lora(lora_scale)
|
114 |
|
115 |
# Set random seed for reproducibility
|
116 |
with calculateDuration("Randomizing seed"):
|
|
|
125 |
final_image = image
|
126 |
yield image, seed # Yield intermediate images and seed
|
127 |
|
128 |
+
with calculateDuration("Unload"):
|
129 |
+
#pipe.unfuse_lora()
|
130 |
pipe.unload_lora_weights()
|
131 |
|
132 |
return final_image, seed # Return the final image and seed
|