torch>=2.0
numpy>=1.24
Pillow>=9.0
tqdm>=4.60
pyyaml>=6.0
transformers>=4.30

[dev]
pytest>=7.4
pytest-cov>=4.1
ruff>=0.1
mypy>=1.5
pre-commit>=3.4

[full]
diffusers>=0.20
opacus>=1.3
datasets>=2.14
scikit-learn>=1.2
matplotlib>=3.7
seaborn>=0.12
wandb>=0.15
peft>=0.5
huggingface_hub>=0.20

[hub]
huggingface_hub>=0.20
datasets>=2.14
