Enhance LoRA loading process in app.py by adding adapter names and setting equal weights for improved functionality

app.py

@@ -51,28 +51,32 @@ scheduler = FlowMatchEulerDiscreteScheduler.from_config(scheduler_config)
 pipe_stage1 = QwenImageEditPlusPipeline.from_pretrained("Qwen/Qwen-Image-Edit-2511",
                                                  scheduler=scheduler,
                                                  torch_dtype=dtype).to(device)
-# Load 4-step Lightning LoRA and fuse
+# Load 4-step Lightning LoRA
 pipe_stage1.load_lora_weights(
         "lightx2v/Qwen-Image-Edit-2511-Lightning",
-        weight_name="Qwen-Image-Edit-2511-Lightning-4steps-V1.0-bf16.safetensors"
+        weight_name="Qwen-Image-Edit-2511-Lightning-4steps-V1.0-bf16.safetensors",
+        adapter_name="lightning"
 )
-pipe_stage1.fuse_lora()
-# Load Stage 1 LoRA and fuse
-pipe_stage1.load_lora_weights(STAGE1_LORA_REPO, weight_name=STAGE1_LORA_WEIGHT)
+# Load Stage 1 LoRA
+pipe_stage1.load_lora_weights(STAGE1_LORA_REPO, weight_name=STAGE1_LORA_WEIGHT, adapter_name="stage1")
+# Set both adapters with equal weights and fuse
+pipe_stage1.set_adapters(["lightning", "stage1"], adapter_weights=[1.0, 1.0])
 pipe_stage1.fuse_lora()
 
 # Load Stage 2 pipeline (Base Body -> Guide Body)
 pipe_stage2 = QwenImageEditPlusPipeline.from_pretrained("Qwen/Qwen-Image-Edit-2511",
                                                  scheduler=scheduler,
                                                  torch_dtype=dtype).to(device)
-# Load 4-step Lightning LoRA and fuse
+# Load 4-step Lightning LoRA
 pipe_stage2.load_lora_weights(
         "lightx2v/Qwen-Image-Edit-2511-Lightning",
-        weight_name="Qwen-Image-Edit-2511-Lightning-4steps-V1.0-bf16.safetensors"
+        weight_name="Qwen-Image-Edit-2511-Lightning-4steps-V1.0-bf16.safetensors",
+        adapter_name="lightning"
 )
-pipe_stage2.fuse_lora()
-# Load Stage 2 LoRA and fuse
-pipe_stage2.load_lora_weights(STAGE2_LORA_REPO, weight_name=STAGE2_LORA_WEIGHT)
+# Load Stage 2 LoRA
+pipe_stage2.load_lora_weights(STAGE2_LORA_REPO, weight_name=STAGE2_LORA_WEIGHT, adapter_name="stage2")
+# Set both adapters with equal weights and fuse
+pipe_stage2.set_adapters(["lightning", "stage2"], adapter_weights=[1.0, 1.0])
 pipe_stage2.fuse_lora()
 
 # # Apply the same optimizations from the first version