transformers
datasets
torch
faiss-cpu