torch>=2.0
pillow>=9.1
pytorch-msssim>=1.0
safetensors>=0.4

[:sys_platform != "darwin" or platform_machine != "x86_64"]
numpy>=1.24

[:sys_platform == "darwin" and platform_machine == "x86_64"]
numpy<2,>=1.24

[test]
pytest>=8
scipy

[train]
torchvision
tensorboard
matplotlib
scipy
tqdm
