transformers
datasets
numpy==1.21.6
torch==1.12.1
torchvision
sentence_transformers
faiss-cpu