accelerate<2.0.0,>=1.6.0
jsonargparse[signatures]<5.0.0,>=4.28.0
lightning<3.0.0,>=2.4.0
pyyaml<7.0,>=6.0
transformers<5.0.0,>=4.44.0
torch<3.0.0,>=2.4.0
torchmetrics<2.0.0,>=1.4.0.post0
wandb<0.19.0,>=0.18.0
