argparse
numpy
python-Levenshtein==0.12.2
scikit-learn
torch>=1.9.1
transformers==4.17.0
sentencepiece