Spaces:
Sleeping
Sleeping
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
|