pytorch-lightning>=1.6.0
torch>=1.10.0
numpy
tqdm

#metrics
torchmetrics>=0.7.0

# huggingface
transformers>=4.11.2
datasets

# task-specific
sentencepiece
torchmetrics[text]
Pillow

# todo remove once protobuf fixed
protobuf==3.20.1
