pytorch-lightning>=1.6.0
torch>=1.10.0
numpy
tqdm
torchmetrics>=0.7.0
transformers>=4.11.2
datasets
sentencepiece
torchmetrics[text]
Pillow
protobuf==3.20.1

[extra]
lightning-bolts>=0.5.0
deepspeed
accelerate>=0.8.0
