omegaconf>=2.1
numpy<2.0.0,>=1.20.0
torch>=2.1.2
torchvision>=0.16.2
lightning>=2.0
six>=1.9
einops>=0.4.0
