pytorch-lightning>=1.7.0
transformers>=4.21.0
datasets>2.0.0
sentencepiece>=0.1.96
torchmetrics[text]>=0.8.0
Pillow>=9.0.0

[extra]
lightning-bolts>=0.6.0
deepspeed>=0.6.0
accelerate>=0.11.0
