AlekseyCalvin commited on
Commit
5a37aa9
·
verified ·
1 Parent(s): bcad601

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -135,9 +135,9 @@ def run_lora(prompt, cfg_scale, steps, selected_index, randomize_seed, seed, wid
135
  # Load LoRA weights
136
  with calculateDuration(f"Loading LoRA weights for {selected_lora['title']}"):
137
  if "weights" in selected_lora:
138
- pipe.load_lora_weights(lora_path, weight_name=selected_lora["weights"], adapter_name="choice")
139
  pipe.load_lora_weights("ostris/OpenFLUX.1", weight_name="openflux1-v0.1.0-fast-lora.safetensors", adapter_name="fast")
140
- pipe.set_adapters(["fast", "choice"], adapter_weights=[1.0, lora_scale])
141
  else:
142
  pipe.load_lora_weights(lora_path, adapter_name="choice")
143
  pipe.load_lora_weights("ostris/OpenFLUX.1", weight_name="openflux1-v0.1.0-fast-lora.safetensors", adapter_name="fast")
 
135
  # Load LoRA weights
136
  with calculateDuration(f"Loading LoRA weights for {selected_lora['title']}"):
137
  if "weights" in selected_lora:
138
+ pipe.load_lora_weights(lora_path, weight_name=selected_lora["weights"], adapter_name="soonr")
139
  pipe.load_lora_weights("ostris/OpenFLUX.1", weight_name="openflux1-v0.1.0-fast-lora.safetensors", adapter_name="fast")
140
+ pipe.set_adapters(["fast", "soonr"], adapter_weights=[1.0, lora_scale])
141
  else:
142
  pipe.load_lora_weights(lora_path, adapter_name="choice")
143
  pipe.load_lora_weights("ostris/OpenFLUX.1", weight_name="openflux1-v0.1.0-fast-lora.safetensors", adapter_name="fast")