transformers~=4.51.0
accelerate==1.8.1
optimum~=1.24.0
huggingface_hub>=0.29.0
numpy<=1.25.2,>=1.22.2
protobuf<4,>=3.20.3

[diffusers]
diffusers<=0.34.0,>=0.31.0
peft==0.16.0

[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.19.8089.0
torch-neuronx==2.7.0.2.8.6734+ac864f72
torch==2.7.0.*
torchvision==0.22.*
neuronx_distributed==0.13.14393
libneuronxla==2.2.4410.0

[quality]
ruff
isort

[sentence-transformers]
sentence-transformers>=2.2.0

[tests]
pytest<=8.0.0
pytest-rerunfailures
cloudpickle
pytest-timeout
pytest-order
psutil
parameterized
GitPython
sentencepiece
datasets
sacremoses
diffusers<=0.34.0,>=0.31.0
safetensors
sentence-transformers>=2.2.0
peft==0.16.0
trl==0.11.4
compel
rjieba
soundfile
librosa
opencv-python-headless
controlnet-aux
mediapipe
timm>=1.0.0
hf_transfer
torchcodec

[training]
trl==0.11.4
peft==0.16.0
evaluate==0.4.3
neuronx_distributed==0.13.14393

[vllm]
vllm==0.9.2
