accelerate

# cannot be installed on mac
# triton

bitsandbytes
# Pin datasets to 3.6.0 to avoid VideoDecoder issues with 4.0.0
datasets==3.6.0
diffusers>=0.32.1

# something broke in transformers > 4.55.4
transformers==4.55.4

huggingface_hub
hf_transfer>=0.1.8
peft>=0.13.0
#decord>=0.6.0
wandb
pandas
torch==2.6.0
torchvision==0.21.0
torchdata==0.10.1
torchao==0.9.0
sentencepiece>=0.2.0
imageio-ffmpeg>=0.5.1
numpy>=1.26.4
kornia>=0.7.3
ruff==0.9.10