transformers<4.39.0,>=4.38.0
optimum
torch
accelerate<0.28.0
diffusers<0.27.0,>=0.26.0

[quality]
ruff
hf_doc_builder

[tests]
pytest<8.0.0
psutil
parameterized
GitPython
optuna
sentencepiece
datasets
safetensors
