datasets<=4.0.0,>=2.16.0
accelerate<=1.11.0,>=1.3.0
peft<=0.17.2,>=0.14.0
trl<=0.9.6,>=0.8.6
gradio<=5.45.0,>=4.38.0
matplotlib>=3.7.0
tyro<0.9.0
spaces
einops
numpy<2.0.0
pandas>=2.0.0
scipy
evaluate
codebleu
sentencepiece
tiktoken
modelscope>=1.14.0
hf-transfer
safetensors<=0.5.3
fire
omegaconf
packaging
protobuf
pyyaml
pydantic<=2.10.6
aenum
uvicorn
fastapi
sse-starlette
av
librosa
adapters

[:python_version < "3.10"]
transformers!=4.52.0,<=4.56.2,>=4.49.0

[:python_version >= "3.10"]
transformers!=4.52.0,<=4.57.0,>=4.49.0

[adam-mini]
adam-mini

[apollo]
apollo-torch

[aqlm]
aqlm[gpu]>=1.1.0

[badam]
badam>=1.2.1

[bitsandbytes]
bitsandbytes>=0.39.0

[deepspeed]
deepspeed<=0.16.9,>=0.10.0

[dev]
pre-commit
ruff
pytest
build

[eetq]
eetq

[fp8]
torchao>=0.8.0
accelerate>=1.10.0

[fp8-all]
torchao>=0.8.0
transformer_engine[pytorch]>=2.0.0
accelerate>=1.10.0

[fp8-te]
transformer_engine[pytorch]>=2.0.0
accelerate>=1.10.0

[galore]
galore-torch

[gptq]
optimum>=1.24.0
gptqmodel>=2.0.0

[hqq]
hqq

[liger-kernel]
liger-kernel>=0.5.5

[metrics]
nltk
jieba
rouge-chinese

[minicpm_v]
soundfile
torchvision
torchaudio
vector_quantize_pytorch
vocos
msgpack
referencing
jsonschema_specifications

[openmind]
openmind

[sglang]
sglang[srt]>=0.4.5
transformers==4.51.1

[swanlab]
swanlab

[torch]
torch>=2.0.0
torchvision>=0.15.0

[torch-npu]
torch-npu==2.5.1
torchvision==0.20.1
decorator

[vllm]
vllm<=0.10.2,>=0.4.3
