loguru
pyyaml>=5.0.0
numpy<2.0,>=1.17.0
requests>=2.0.0
tqdm>=4.0.0
torch>=1.7.0
transformers<5.0,>4.0
datasets
accelerate!=1.1.0,>=0.20.3
pynvml
pillow
compressed-tensors>=0.9.4a2

[dev]
pytest>=6.0.0
pytest-mock>=3.6.0
pytest-rerunfailures>=13.0
parameterized
lm_eval==0.4.5
beautifulsoup4~=4.12.3
cmarkgfm~=2024.1.14
trl>=0.10.1
pandas
black~=24.4.2
isort~=5.13.2
mypy~=1.10.0
ruff~=0.4.8
flake8~=7.0.0
pre-commit
