Update app.py

app.py

@@ -32,31 +32,46 @@ pipe = QwenImageEditPlusPipeline.from_pretrained(
     torch_dtype=dtype
 ).to(device)
 
-# Load first LoRA
-pipe.load_lora_weights(
-    "dx8152/Qwen-Edit-2509-Light-Migration",
-    weight_name="参考色调.safetensors",
-    adapter_name="light_migration"
-)
+scheduler_config = {
+    "base_image_seq_len": 256,
+    "base_shift": math.log(3),  
+    "invert_sigmas": False,
+    "max_image_seq_len": 8192,
+    "max_shift": math.log(3),  
+    "num_train_timesteps": 1000,
+    "shift": 1.0,
+    "shift_terminal": None,  
+    "stochastic_sampling": False,
+    "time_shift_type": "exponential",
+    "use_beta_sigmas": False,
+    "use_dynamic_shifting": True,
+    "use_exponential_sigmas": False,
+    "use_karras_sigmas": False,
+}
+
+scheduler = FlowMatchEulerDiscreteScheduler.from_config(scheduler_config)
+
+pipe = QwenImageEditPlusPipeline.from_pretrained(
+    "Qwen/Qwen-Image-Edit-2509",
+    scheduler=scheduler,
+    torch_dtype=dtype
+).to(device)
 
-# Load second LoRA
 pipe.load_lora_weights(
     "lightx2v/Qwen-Image-Lightning",
-    weight_name="Qwen-Image-Lightning-8steps-V2.0-bf16.safetensors",
-    adapter_name="lightning"
+    weight_name="Qwen-Image-Lightning-8steps-V2.0.safetensors",
+    adapter_name="fast"
 )
 
-# Set both adapters active with their weights
-pipe.set_adapters(
-    ["light_migration", "lightning"], 
-    adapter_weights=[1.0, 1.0]
+pipe.load_lora_weights(
+    "dx8152/Qwen-Edit-2509-Light-Migration",
+    weight_name="参考色调.safetensors",
+    adapter_name="angles"
 )
 
-# Fuse both LoRAs (adjust scales as needed)
-pipe.fuse_lora(
-    adapter_names=["light_migration", "lightning"], 
-    lora_scale=1.0
-)
+pipe.set_adapters(["angles","fast"], adapter_weights=[1.,1.])
+pipe.fuse_lora(adapter_names=["angles"], lora_scale=1.)
+pipe.fuse_lora(adapter_names=["fast"], lora_scale=1.)
 pipe.unload_lora_weights()
 
 #spaces.aoti_blocks_load(pipe.transformer, "zerogpu-aoti/Qwen-Image", variant="fa3")
@@ -75,7 +90,6 @@ MAX_SEED = np.iinfo(np.int32).max
 # Default prompt for light migration
 DEFAULT_PROMPT = "参考色调，移除图1原有的光照并参考图2的光照和色调对图1重新照明"
 
-
 @spaces.GPU
 def infer_light_migration(
     image: Optional[Image.Image] = None,