Update app.py

app.py

@@ -14,6 +14,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
 
 # --- Prompt Enhancement using Hugging Face InferenceClient ---
 def polish_prompt_hf(original_prompt, system_prompt):
@@ -164,6 +165,7 @@ scheduler = FlowMatchEulerDiscreteScheduler.from_config(scheduler_config)
 
 pipe = QwenImageEditPipeline.from_pretrained("Qwen/Qwen-Image-Edit", scheduler=scheduler,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")