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"):
|