Pre-Compiled AoTI (#21)

- Pre-Compiled AoTI (f0dfdd22d5bce950dae8826ef43b375a7f700ba0)
- Update requirements.txt (f04829345d61a073405442ddc260c45b7482edf8)

app.py

@@ -186,12 +186,14 @@ except Exception as e:
     print(f"Warning: Could not load Lightning LoRA weights: {e}")
     print("Continuing with base model...")
 
+spaces.aoti_blocks_load(pipe.transformer, "zerogpu-aoti/Qwen-Image", variant="fa3")
+
 # Apply the same optimizations from the first version
-pipe.transformer.__class__ = QwenImageTransformer2DModel
-pipe.transformer.set_attn_processor(QwenDoubleStreamAttnProcessorFA3())
+# 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")
+# optimize_pipeline_(pipe, image=Image.new("RGB", (1024, 1024)), prompt="prompt")
 
 # --- UI Constants and Helpers ---
 MAX_SEED = np.iinfo(np.int32).max

requirements.txt

@@ -7,4 +7,5 @@ safetensors
 sentencepiece
 dashscope
 torchvision
-peft
+peft
+torch==2.8