PyYAML<7.0.0,>=6.0.2
jsonargparse[signatures]<5.0.0,>=4.40.0
lightning==2.6.0
omegaconf==2.3.0
torch==2.9.1
torchmetrics==1.8.2
transformers<5.0.0,>=4.44.0
yoyodyne>=0.5.12
wandb<0.19.8,>=0.18.7
