matrix-game-2 / requirements.txt
dn6's picture
dn6 HF Staff
Upload folder using huggingface_hub
e0e2430 verified
raw
history blame contribute delete
465 Bytes
accelerate==1.10.1
einops==0.8.1
flash-attn @ https://github.com/Dao-AILab/flash-attention/releases/download/v2.7.4.post1/flash_attn-2.7.4.post1+cu12torch2.6cxx11abiTRUE-cp310-cp310-linux_x86_64.whl
hf-transfer==0.1.9
hf-xet==1.1.8
huggingface-hub==0.34.4
imageio==2.37.0
imageio-ffmpeg==0.6.0
safetensors==0.6.2
sentencepiece==0.2.1
torch==2.7.1
torchao==0.12.0
torchvision==0.22.1
transformers==4.55.4
diffusers @ git+https://github.com/huggingface/diffusers.git