Update app.py

app.py

@@ -12,6 +12,7 @@ import math
 from optimization import optimize_pipeline_
 from qwenimage.pipeline_qwen_image_edit import QwenImageEditPipeline
 from qwenimage.transformer_qwenimage import QwenImageTransformer2DModel
+from qwenimage.qwen_fa3_processor import QwenDoubleStreamAttnProcessorFA3
 
 from huggingface_hub import InferenceClient
 import math
@@ -143,6 +144,8 @@ dtype = torch.bfloat16
 device = "cuda" if torch.cuda.is_available() else "cpu"
 pipe = QwenImageEditPipeline.from_pretrained("Qwen/Qwen-Image-Edit", torch_dtype=dtype).to(device)
 pipe.transformer.__class__ = QwenImageTransformer2DModel
+pipe.transformer.set_attn_processor(QwenDoubleStreamAttnProcessorFA3())
+
 
 # --- Ahead-of-time compilation ---
 optimize_pipeline_(pipe, image=Image.new("RGB", (1024, 1024)), prompt="prompt")