torch
sentencepiece
transformers
textgen==1.0.1