accelerate>=1.13.0
datasets>=4.7.0
huggingface-hub>=0.36.2
matplotlib>=3.10.8
numpy>=2.4.4
peft>=0.18.1
rouge>=1.0.1
safetensors>=0.7.0
scikit-learn>=1.8.0
scipy>=1.17.1
torch>=2.11.0
tqdm>=4.67.3
transformers>=4.57.6

[audio]
torchaudio>=2.11.0
librosa>=0.11.0
soundfile>=0.13.1

[datasets]
modelscope>=1.9
gdown>=4.0

[dev]
detectzoo[audio,datasets,image]
pytest>=7.0
pytest-cov>=4.0
ruff>=0.15.5

[image]
torchvision>=0.26.0
Pillow>=12.1.1
open-clip-torch>=2.20
diffusers>=0.21
lpips>=0.1.4
pytorch-wavelets>=1.3
timm>=0.9.0
