requirements.txt

accelerate==0.15.0
datasets==2.11.0
evaluate==0.3.0
nltk==3.7
numpy==1.23.5
omegaconf==2.3.0
parallelformers==1.2.7
promptsource==0.2.3
rouge-score==0.1.2
scikit-learn==1.1.3
setuptools==61.2.0
tqdm==4.64.0
transformers[sentencepiece]==4.27.1