numpy
transformers==4.14.1
torch
accelerate
nltk