torch>=1.9.0
torchvision>=0.10.0
numpy>=1.21.0
einops>=0.4.0
omegaconf>=2.1.0
scipy>=1.7.0
PyWavelets>=1.1.0

[alignment]
opencv-python-headless>=4.5.0
open-clip-torch>=2.0.0

[all]
timm>=0.6.0
monai>=1.0.0
opencv-python-headless>=4.5.0
open-clip-torch>=2.0.0
pytorch-wavelets>=1.3.0
accelerate>=0.20.0
transformers>=4.30.0
packaging>=21.0
memory-efficient-attention-pytorch>=0.1.0
diffusers>=0.20.0
Pillow>=9.0.0

[dev]
pytest>=7.0.0
pytest-cov>=4.0.0
black>=23.0.0
flake8>=6.0.0
mypy>=1.0.0
build>=1.0.0
twine>=5.1.0

[diffusers]
diffusers>=0.20.0

[distributed]
accelerate>=0.20.0

[monai]
monai>=1.0.0

[muse]
transformers>=4.30.0
packaging>=21.0
memory-efficient-attention-pytorch>=0.1.0

[pillow]
Pillow>=9.0.0

[timm]
timm>=0.6.0

[wavelets]
pytorch-wavelets>=1.3.0
