blobfile>=3.0.0
bytecheckpoint
datasets<=2.21.0,>=2.16.0
packaging<26.0,>=23.0
torchdata<1.0,>=0.8.0
tiktoken>=0.9.0
diffusers<=0.31.0,>=0.30.0
packaging<26.0,>=23.0
transformers[torch]<4.57.0,>=4.51.3

[audio]
av<15.0,>=14.3.0
librosa<0.12,>=0.11.0
soundfile<0.14,>=0.13.1

[dev]
pre-commit<5.0,>=4.0.0
ruff<1.0,>=0.7.0
pytest<8.0,>=6.0.0
expecttest<0.4,>=0.3.0

[dit]
diffusers<=0.31.0,>=0.30.0
bitsandbytes<=0.47.0,>=0.46.0

[gpu]
liger-kernel<=0.5.6,>=0.5.5
flash-attn==2.6.3

[npu]
torchvision<0.16.1,>=0.16.0
