Update app.py

app.py

@@ -7,9 +7,7 @@ import spaces
 from PIL import Image
 from diffusers import FlowMatchEulerDiscreteScheduler
 from optimization import optimize_pipeline_
-from qwenimage.pipeline_qwenimage_edit_plus import QwenImageEditPlusPipeline
-from qwenimage.transformer_qwenimage import QwenImageTransformer2DModel
-from qwenimage.qwen_fa3_processor import QwenDoubleStreamAttnProcessorFA3
+from diffusers import QwenImageEditPlusPipeline
 
 import math
 from huggingface_hub import hf_hub_download
@@ -27,11 +25,6 @@ from typing import Optional, Tuple, Any
 dtype = torch.bfloat16
 device = "cuda" if torch.cuda.is_available() else "cpu"
 
-pipe = QwenImageEditPlusPipeline.from_pretrained(
-    "Qwen/Qwen-Image-Edit-2509",
-    torch_dtype=dtype
-).to(device)
-
 scheduler_config = {
     "base_image_seq_len": 256,
     "base_shift": math.log(3),  
@@ -77,8 +70,7 @@ pipe.unload_lora_weights()
 
 #spaces.aoti_blocks_load(pipe.transformer, "zerogpu-aoti/Qwen-Image", variant="fa3")
 
-pipe.transformer.__class__ = QwenImageTransformer2DModel
-pipe.transformer.set_attn_processor(QwenDoubleStreamAttnProcessorFA3())
+pipe.transformer.set_attention_backend("_flash_3_hub")
 
 optimize_pipeline_(
     pipe,