sentence_transformers
pymongo
transformers
torch
accelerate
datasets