pytorch-lightning==2.0.0
torchmetrics==0.11.0
torch==2.0.1
omegaconf==2.2.3
