supervision~=0.24.0rc1
requests<=2.32.3,>=2.31.0
transformers~=4.44.2
torch~=2.4.0
accelerate~=0.33.0
sentencepiece~=0.2.0
peft~=0.12.0
flash-attn~=2.6.3
einops~=0.8.0
timm~=1.0.9
typer~=0.12.5

[dev]
pytest~=8.3.2
black~=24.8.0
pre-commit~=3.8.0
mypy~=1.11.2
flake8~=7.1.1

[docs]
mkdocs-material~=9.5.33
mkdocstrings[python]<0.25.2,>=0.20.0
