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

[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<2.3.0
torchvision
librosa
soundfile
torchcodec
black~=24.4.2
isort~=5.13.2
mypy~=1.10.0
ruff~=0.4.8
flake8~=7.0.0
pre-commit
mkdocs
mkdocs-material[imaging]
markdown
pymdown-extensions
mkdocs-section-index
mkdocs-minify-plugin
mkdocs-api-autonav
mkdocstrings-python
mkdocs-gen-files
mkdocs-nav-weight
