datasets
faiss-cpu
faiss-gpu
transformers
torch