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

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