loguru<=0.7.3,>=0.7.2
pyyaml<=6.0.2,>=6.0.1
numpy<=2.3.2,>=2.0.0
requests<=2.32.5,>=2.32.2
tqdm<=4.67.1,>=4.66.3
torch<=2.8.0,>=2.7.0
transformers<=4.55.2,>=4.53.0
datasets<=4.0.0,>=4.0.0
accelerate<=1.10.0,>=1.6.0
pynvml<=12.0.0,>=11.5.3
pillow<=12.1.1,>=10.4.0
compressed-tensors==0.11.0

[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==0.11.0
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
