transformers<5.0,>=4.42.0
huggingface-hub<1.0,>=0.24.0
typer<1.0,>=0.12.0
pyyaml<7.0,>=6.0
rich<15.0,>=13.0.0
prompt-toolkit<4.0,>=3.0.0
regex>=2024.0.0
accelerate<1.0,>=0.25.0

[cpu]
torch<3.0,>=2.0.0

[cuda]
torch<3.0,>=2.0.0
bitsandbytes<1.0,>=0.41.0

[dev]
pytest<9.0,>=7.0.0
pytest-cov<8.0,>=4.0.0
black>=23.0.0
ruff>=0.1.0

[mlx]
mlx>=0.22.0
mlx-lm>=0.20.0
