fire
sentencepiece
torch>=2.1.0
xformers
simple-parsing