transformers<4.52.0,>=4.51.0
optimum~=2.0
torch
accelerate>=1.7.0
diffusers<0.34.1,>=0.34.0
huggingface_hub[hf_xet]>=0.24.7
sentence-transformers==3.3.1
sentencepiece

[quality]
ruff
hf_doc_builder

[tests]
psutil
parameterized
GitPython
optuna
sentencepiece
datasets==3.6.0
timm
safetensors
pytest<8.0.0
scipy
torchsde
peft
bitsandbytes
tiktoken
blobfile
