transformers<4.52.0,>=4.51.0
optimum~=1.25
torch
accelerate>=1.7.0
diffusers<0.33.2,>=0.33.1
huggingface_hub[hf_xet]>=0.24.7
sentence-transformers==3.3.1

[quality]
ruff
hf_doc_builder

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