torch<2.8.0,>=2.6.0
transformers<5.0.0,>=4.48.0
datasets<5.0.0,>=4.0.0
dataclasses_json

[all]
llmcompressor==0.5.1
lightning<3.0.0,>=2.3.2
liger-kernel<1.0.0,>=0.5.0

[dev]
matplotlib>=3.7.2
flake8>=4.0.1.1
black>=24.4.2
isort>=5.13.2
pytest>=7.1.2
pre-commit
pytest-xdist
pytest-rerunfailures
seaborn
mkdocs
mkdocs-material
llmcompressor==0.5.1
lightning<3.0.0,>=2.3.2
liger-kernel<1.0.0,>=0.5.0

[inference]
llmcompressor==0.5.1

[train]
lightning<3.0.0,>=2.3.2
liger-kernel<1.0.0,>=0.5.0
