Update app.py

app.py

@@ -151,12 +151,15 @@ def run_lora(prompt, negative_prompt, cfg_scale, steps, selected_index, seed, wi
     # Load LoRA weights
     with calculateDuration(f"Loading LoRA weights for {selected_lora['title']}"):
         if "weights" in selected_lora:
-            pipe.load_lora_weights("ostris/OpenFLUX.1", weight_name="openflux1-v0.1.0-fast-lora.safetensors", adapter_weights=[1.0])
-            pipe.load_lora_weights(lora_path, weight_name=selected_lora["weights"], adapter_name=selected_lora["repo"], adapter_weights=[lora_scale])
+            pipe.load_lora_weights("ostris/OpenFLUX.1", weight_name="openflux1-v0.1.0-fast-lora.safetensors", adapter_weights=[1.0], lora_scale=[1.0])
+            pipe.load_lora_weights(lora_path, weight_name=selected_lora["weights"], adapter_name=selected_lora["repo"], lora_scale=[1.0], adapter_weights=selected_lora[lora_scale])
+            pipe.set_adapters(["fast", selected_lora["repo"]], adapter_weights=[1.0, selected_lora[lora_scale]])
         else:
-            pipe.load_lora_weights("ostris/OpenFLUX.1", weight_name="openflux1-v0.1.0-fast-lora.safetensors", adapter_weights=[1.0])
-            pipe.load_lora_weights(lora_path, adapter_name=selected_lora["repo"], adapter_weights=[lora_scale])
-    
+            pipe.load_lora_weights("ostris/OpenFLUX.1", weight_name="openflux1-v0.1.0-fast-lora.safetensors", adapter_weights=[1.0], lora_scale=[1.0])
+            pipe.load_lora_weights(lora_path, adapter_name=selected_lora["repo"], adapter_weights=selected_lora[lora_scale])
+            pipe.set_adapters(["fast", selected_lora["repo"]], adapter_weights=[1.0, selected_lora[lora_scale]])
+
+
     image = generate_image(prompt, negative_prompt, steps, seed, cfg_scale, width, height, lora_scale)
     pipe.to("cpu")
     pipe.unload_lora_weights()