pyarrow>=14
huggingface_hub>=0.34
transformers>=4.56
accelerate>=0.26
optuna>=4.0
gguf>=0.19
rich>=13.0
nvidia-ml-py>=12.535.108
sentencepiece<0.3.0,>=0.1.98
senbonzakura-check<0.5,>=0.4.0

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

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

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

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

[abliterate]
senbonzakura

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

[completion]
shtab>=1.6

[dev]
senbonzakura[abliterate,completion,hub]
pytest>=7
pytest-cov>=4
pytest-xdist>=3.2
build>=1.0
ruff==0.16.0

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

[hub]
datasets>=2.15

[quant]
bitsandbytes>=0.43
