torch
sentencepiece
pyyaml
tqdm