text-generation==0.6.0 transformers==4.31.0 pymongo==4.4.1 torch sentencepiece protobuf datasets==2.14.6