transformers datasets numpy==1.21.6 torch>=2.6 torchvision sentence_transformers faiss-cpu