torch>=2.2.2
torchvision>=0.17.2
numpy>=1.24.4
setuptools>=65.5.0
einops>=0.8.0
opencv-python>=4.9.0.80
efficientnet_pytorch==0.7.1
vit-pytorch==1.6.9
platformdirs
gdown
diffusers
matplotlib
wandb

[test]
pytest
coverage
flake8
black
isort
pytest-cov
mypy
gitchangelog
mkdocs
lark
types-PyYAML
