accelerate>=1.4.0
datasets>=3.0.0
transformers>=4.55.0

[bco]
scikit-learn
joblib

[deepspeed]
deepspeed>=0.14.4

[dev]
scikit-learn
joblib
deepspeed>=0.14.4
diffusers>=0.18.0
ftfy
openai>=1.23.2
llm-blender>=0.0.2
liger-kernel>=0.6.2
peft>=0.8.0
bitsandbytes
parameterized
pytest-cov
pytest-rerunfailures==15.1
pytest-xdist
pytest
Pillow
torchvision
num2words

[diffusers]
diffusers>=0.18.0
ftfy

[judges]
openai>=1.23.2
llm-blender>=0.0.2

[liger]
liger-kernel>=0.6.2

[peft]
peft>=0.8.0

[quantization]
bitsandbytes

[scikit]
scikit-learn

[test]
parameterized
pytest-cov
pytest-rerunfailures==15.1
pytest-xdist
pytest

[vllm]

[vllm:python_version < "3.13"]
vllm>=0.10.0
fastapi
pydantic
requests
uvicorn

[vlm]
Pillow
torchvision
num2words
