requirements.txt

# Mage-Flow Space requirements
# Do NOT list gradio, spaces, or huggingface_hub (preinstalled on the platform).

torchvision
diffusers>=0.37.0
transformers>=5.3.0,<5.6
accelerate>=1.0.0
safetensors>=0.4.0
einops>=0.8.0
pydantic>=2.0
pillow>=10.0
numpy>=1.26
loguru>=0.7.0
typing_extensions>=4.0

# Flash Attention 2 (prebuilt Blackwell sm_120 wheel)
https://huggingface.co/datasets/multimodalart/zerogpu-blackwell-wheels/resolve/main/wheels/pt211-cu130-cp312/flash_attn-2.8.3-cp312-cp312-linux_x86_64.whl