Update app.py

app.py

@@ -49,16 +49,16 @@ pipe.set_adapters(["light_migration"], adapter_weights=[1.])
 pipe.fuse_lora(adapter_names=["light_migration"], lora_scale=1.25)
 pipe.unload_lora_weights()
 
-spaces.aoti_blocks_load(pipe.transformer, "zerogpu-aoti/Qwen-Image", variant="fa3")
+#spaces.aoti_blocks_load(pipe.transformer, "zerogpu-aoti/Qwen-Image", variant="fa3")
 
-#pipe.transformer.__class__ = QwenImageTransformer2DModel
-#pipe.transformer.set_attn_processor(QwenDoubleStreamAttnProcessorFA3())
+pipe.transformer.__class__ = QwenImageTransformer2DModel
+pipe.transformer.set_attn_processor(QwenDoubleStreamAttnProcessorFA3())
 
-#optimize_pipeline_(
-#    pipe,
-#    image=[Image.new("RGB", (1024, 1024)), Image.new("RGB", (1024, 1024))],
-#    prompt="prompt"
-#)
+optimize_pipeline_(
+    pipe,
+    image=[Image.new("RGB", (1024, 1024)), Image.new("RGB", (1024, 1024))],
+    prompt="prompt"
+)
 
 MAX_SEED = np.iinfo(np.int32).max