# FlashAttention-2 for sm_120 (Blackwell). MANDATORY: cosmos_framework's attention
# dispatcher offers only ["flash2", "natten"] on sm>=80 -- there is no SDPA fallback.
# torch is pinned to 2.11.0 so the prebuilt wheel's cell matches exactly.
torch==2.11.0
torchvision==0.26.0
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

# The authors' inference stack (NVIDIA cosmos-framework fork shipped with SyncWorld).
# Pinned to the commit the checkpoint was released with.
cosmos-framework @ git+https://github.com/UMass-Embodied-AGI/SyncWorld.git@f026490eb5b15640087b575e13952c86c3d9b7b3

# Imported by the modules the demo touches but not declared by cosmos-framework's
# base extra (they live in its `train` / `dev` extras, which also pull gradio).
attrs
einx
iopath
qwen-vl-utils
multi-storage-client==0.44.0
boto3
pandas
pyarrow
matplotlib
opencv-python-headless
mediapy
ftfy
webdataset
scikit-image
tensorstore
zarr
h5py
lz4
more-itertools
wandb
imageio
imageio-ffmpeg
av