Spaces:
Running
on
Zero
Running
on
Zero
Commit
•
ac0abd9
1
Parent(s):
bf027a7
Update app.py
Browse files
app.py
CHANGED
@@ -145,7 +145,7 @@ def run_lora(prompt, image_input, image_strength, cfg_scale, steps, selected_ind
|
|
145 |
pipe_to_use.load_lora_weights(
|
146 |
lora_path,
|
147 |
weight_name=weight_name,
|
148 |
-
low_cpu_mem_usage=
|
149 |
)
|
150 |
|
151 |
# Set random seed for reproducibility
|
|
|
145 |
pipe_to_use.load_lora_weights(
|
146 |
lora_path,
|
147 |
weight_name=weight_name,
|
148 |
+
low_cpu_mem_usage=True
|
149 |
)
|
150 |
|
151 |
# Set random seed for reproducibility
|