Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -231,8 +231,10 @@ def run_lora(prompt, cfg_scale, steps, selected_index, randomize_seed, seed, asp
|
|
231 |
pipe.load_lora_weights(
|
232 |
lora_path,
|
233 |
weight_name=weight_name,
|
234 |
-
low_cpu_mem_usage=True
|
|
|
235 |
)
|
|
|
236 |
|
237 |
# Set random seed for reproducibility
|
238 |
with calculateDuration("Randomizing seed"):
|
|
|
231 |
pipe.load_lora_weights(
|
232 |
lora_path,
|
233 |
weight_name=weight_name,
|
234 |
+
low_cpu_mem_usage=True,
|
235 |
+
adapter_name="style"
|
236 |
)
|
237 |
+
pipe.set_adapters(["style"], adapter_weights=[lora_scale])
|
238 |
|
239 |
# Set random seed for reproducibility
|
240 |
with calculateDuration("Randomizing seed"):
|