torch >= 1.9.0
torchaudio >= 0.9.0
torchvision >= 0.10.0
torchmetrics >= 0.7.3, < 0.11.0
lightning >= 2.0, < 3
tensorboard >= 2.0.0
pillow >= 8.1
av >= 8.0.3
pysoundfile >= 0.9.0; platform_system == "Windows"
numpy >= 1.20.1, < 2.0
einops >= 0.3.0
matplotlib >= 3.4.2
pyyaml == 6.*
toml >= 0.10.0
safetensors >= 0.3.1
psycopg == 3.*