pyarrow>=23.0.1
huggingface_hub>=1.5.0
transformers>=5.10.1
accelerate>=0.32
optuna>=4.0
gguf>=0.19
rich>=13.0
nvidia-ml-py>=12.535.108
sentencepiece<0.3.0,>=0.2.1
senbonzakura-check<0.6,>=0.5.1

[:python_version < "3.11"]
tomli>=2.0

[:python_version < "3.13"]
torch>=2.7

[:python_version == "3.13"]
torch>=2.7

[:python_version >= "3.14"]
torch>=2.9

[abliterate]
senbonzakura

[all]
senbonzakura[abliterate,completion,finetune,hub,quant]

[completion]
shtab>=1.6

[dev]
senbonzakura[abliterate,completion,finetune,hub]
pytest>=9.0.3
pytest-cov>=4
pytest-xdist>=3.2
build>=1.0
setuptools-rust==1.13.0
ruff==0.16.0

[dev:python_version < "3.11"]
tomli>=2.0

[finetune]
peft>=0.20.0

[hub]
datasets>=5.0.1
multidict>=6.9.1

[quant]
bitsandbytes>=0.43
