einops>=0.3
faiss-gpu
numpy
torch>=1.6
tqdm
