Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -137,13 +137,13 @@ def run_lora(prompt, cfg_scale, steps, selected_index, randomize_seed, seed, wid
|
|
137 |
# Load LoRA weights
|
138 |
with calculateDuration(f"Loading LoRA weights for {selected_lora['title']}"):
|
139 |
if "weights" in selected_lora:
|
140 |
-
pipe.load_lora_weights(lora_path, weight_name=selected_lora["weights"], adapter_name="
|
141 |
pipe.load_lora_weights("ostris/OpenFLUX.1", weight_name="openflux1-v0.1.0-fast-lora.safetensors", adapter_name="fast")
|
142 |
-
pipe.set_adapters(["fast", "
|
143 |
else:
|
144 |
-
pipe.load_lora_weights(lora_path, adapter_name="
|
145 |
pipe.load_lora_weights("ostris/OpenFLUX.1", weight_name="openflux1-v0.1.0-fast-lora.safetensors", adapter_name="fast")
|
146 |
-
pipe.set_adapters(["fast", "
|
147 |
|
148 |
# Set random seed for reproducibility
|
149 |
with calculateDuration("Randomizing seed"):
|
|
|
137 |
# Load LoRA weights
|
138 |
with calculateDuration(f"Loading LoRA weights for {selected_lora['title']}"):
|
139 |
if "weights" in selected_lora:
|
140 |
+
pipe.load_lora_weights(lora_path, weight_name=selected_lora["weights"], adapter_name="soon")
|
141 |
pipe.load_lora_weights("ostris/OpenFLUX.1", weight_name="openflux1-v0.1.0-fast-lora.safetensors", adapter_name="fast")
|
142 |
+
pipe.set_adapters(["fast", "soon"], adapter_weights=[1.0, lora_scale])
|
143 |
else:
|
144 |
+
pipe.load_lora_weights(lora_path, adapter_name="soon")
|
145 |
pipe.load_lora_weights("ostris/OpenFLUX.1", weight_name="openflux1-v0.1.0-fast-lora.safetensors", adapter_name="fast")
|
146 |
+
pipe.set_adapters(["fast", "soon"], adapter_weights=[1.0, lora_scale])
|
147 |
|
148 |
# Set random seed for reproducibility
|
149 |
with calculateDuration("Randomizing seed"):
|