torch==1.9.0
pytorch-lightning==1.5.9
torchmetrics==0.5.1
transformers==4.11.3
hydra-core==1.1.1
argcomplete==1.12.3
datasets==1.12.1
seqeval==1.2.2
scikit-learn==0.24.2
matplotlib==3.4.3
rich==11.0.0
sentencepiece==0.1.96
wandb==0.12.2
nltk==3.6.5
rouge-score==0.0.4
sacrebleu==2.0.0
