transformers
torch
datasets
faiss-cpu