transformers<4.56.0,>=4.55.0
optimum<2.1.0,>=2.0.0
torch
accelerate<=1.10.1,>=1.7.0
diffusers<0.35.2,>=0.35.1
huggingface_hub[hf_xet]>=0.24.7
sentence-transformers==3.3.1
sentencepiece

[quality]
ruff

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