torch>=2.1
optimum-onnx==0.0.*
transformers<4.56,>=4.45
setuptools

[diffusers]
diffusers

[ipex]
intel-extension-for-pytorch>=2.8
transformers<4.56,>4.54
accelerate

[neural-compressor]
neural-compressor[pt]>=3.4.1
accelerate
transformers<4.46
datasets

[nncf]
nncf>=2.18.0

[openvino]
nncf>=2.18.0
openvino>=2025.1.0
openvino-tokenizers>=2025.1.0

[quality]
black~=23.1
ruff==0.4.4

[tests]
accelerate
pytest<8.0.0,>=7.2.0
parameterized
Pillow
evaluate
py-cpuinfo
sacremoses
sentencepiece
torchaudio
rjieba
timm
invisible-watermark>=0.2.0
transformers_stream_generator
einops
tiktoken
sentence-transformers
open_clip_torch>=2.26.1
peft
datasets[audio]<4.0.0,>=1.4.0
tbb
langchain-huggingface
hf_xet
num2words
vocos
vector_quantize_pytorch
