Update app.py

app.py

@@ -28,8 +28,8 @@ dtype = torch.bfloat16
 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',
+                                                #transformer= QwenImageTransformer2DModel.from_pretrained("linoyts/Qwen-Image-Edit-Rapid-AIO", 
+                                                                                                         # subfolder='transformer',
                                                                                                          torch_dtype=dtype,
                                                                                                          device_map='cuda'),torch_dtype=dtype).to(device)
 
@@ -39,8 +39,10 @@ pipe.load_lora_weights(
         weight_name="Qwen-Edit-Relight.safetensors", adapter_name="relight"
     )
 
-pipe.set_adapters(["relight"], adapter_weights=[1.])
-pipe.fuse_lora(adapter_names=["relight"], lora_scale=1)
+pipe.load_lora_weights("lightx2v/Qwen-Image-Lightning",
+        weight_name="Qwen-Image-Lightning-4steps-V2.0.safetensors", adapter_name="lightning")
+pipe.set_adapters(["relight", "lightning"], adapter_weights=[1., 1.])
+pipe.fuse_lora(adapter_names=["relight", "lightning"], lora_scale=1)
 pipe.unload_lora_weights()
 
 pipe.transformer.__class__ = QwenImageTransformer2DModel