torch>=1.11
optimum==1.25.*
transformers<4.52,>=4.36
datasets>=1.4.0
setuptools
scipy
onnx

[diffusers]
diffusers

[ipex]
intel-extension-for-pytorch>=2.6
transformers<4.52,>4.49
accelerate

[neural-compressor]
neural-compressor[pt]>3.0
accelerate
transformers<4.46

[nncf]
nncf>=2.16.0

[openvino]
nncf>=2.16.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]>=1.4.0
tbb
langchain-huggingface
hf_xet
num2words
