torch>=1.12.0
torchaudio>=0.12.0
sentencepiece
pyyaml
tensorboardX
pillow
termcolor
tabulate
torchviz
editdistance
flashlight-text