transformers~=4.55.4
accelerate==1.8.1
optimum~=1.24.0
huggingface_hub>=0.31.4
numpy<=1.26.4,>=1.22.2
protobuf<4,>=3.20.3

[diffusers]
diffusers==0.35.*
peft==0.17.0

[diffusers-tests]
compel==2.1.1

[neuron]
wheel
torch-neuron==1.13.1.2.9.74.0
torch==1.13.1.*
neuron-cc[tensorflow]==1.22.0.0
protobuf
torchvision
numpy==1.22.3

[neuronx]
wheel
neuronx-cc==2.21.18209.0
torch-neuronx==2.8.0.2.10.13553
torch==2.8.0.*
torchvision==0.23.*
neuronx_distributed==0.15.22404
libneuronxla==2.2.12677.0

[quality]
pre-commit
ruff==0.13.0
isort

[sentence-transformers]
sentence-transformers>=2.2.0

[tests]
pytest<=8.0.0
pytest-rerunfailures
cloudpickle
pytest-timeout
pytest-forked
psutil
parameterized
GitPython
sentencepiece
datasets
sacremoses
safetensors
sentence-transformers>=2.2.0
rjieba
soundfile
librosa
controlnet-aux
torchcodec==0.7.0

[training]
trl==0.11.4
peft==0.17.0
evaluate==0.4.3

[vllm]
vllm==0.10.2

[vllm-tests]
docker
pytest-asyncio
openai
