Update app.py

app.py

@@ -29,24 +29,34 @@ device = "cuda" if torch.cuda.is_available() else "cpu"
 
 pipe = QwenImageEditPlusPipeline.from_pretrained(
     "Qwen/Qwen-Image-Edit-2509",
-    transformer=QwenImageTransformer2DModel.from_pretrained(
-        "linoyts/Qwen-Image-Edit-Rapid-AIO",
-        subfolder='transformer',
-        torch_dtype=dtype,
-        device_map='cuda'
-    ),
     torch_dtype=dtype
 ).to(device)
 
-# Load Light Migration LoRA
+# Load first LoRA
 pipe.load_lora_weights(
     "dx8152/Qwen-Edit-2509-Light-Migration",
     weight_name="参考色调.safetensors",
     adapter_name="light_migration"
 )
 
-pipe.set_adapters(["light_migration"], adapter_weights=[1.])
-pipe.fuse_lora(adapter_names=["light_migration"], lora_scale=1.25)
+# Load second LoRA
+pipe.load_lora_weights(
+    "lightx2v/Qwen-Image-Lightning",
+    weight_name="Qwen-Image-Lightning-8steps-V2.0-bf16.safetensors",
+    adapter_name="lightning"
+)
+
+# Set both adapters active with their weights
+pipe.set_adapters(
+    ["light_migration", "lightning"], 
+    adapter_weights=[1.0, 1.0]
+)
+
+# Fuse both LoRAs (adjust scales as needed)
+pipe.fuse_lora(
+    adapter_names=["light_migration", "lightning"], 
+    lora_scale=1.0
+)
 pipe.unload_lora_weights()
 
 #spaces.aoti_blocks_load(pipe.transformer, "zerogpu-aoti/Qwen-Image", variant="fa3")
@@ -74,10 +84,10 @@ def infer_light_migration(
     seed: int = 0,
     randomize_seed: bool = True,
     true_guidance_scale: float = 1.0,
-    num_inference_steps: int = 4,
+    num_inference_steps: int = 8,
     height: Optional[int] = None,
     width: Optional[int] = None,
-    progress: gr.Progress = gr.Progress(track_tqdm=True)
+    progress: Optional[gr.Progress] = gr.Progress(track_tqdm=True)
 ) -> Tuple[Image.Image, int]:
     """
     Transfer lighting and color tones from a reference image to a source image
@@ -252,7 +262,7 @@ with gr.Blocks() as demo:
                         minimum=1,
                         maximum=40,
                         step=1,
-                        value=4
+                        value=8
                     )
                     height = gr.Slider(
                         label="Height",