From c01ceffeb48157d3b7fd3f44f5a58a553b9f172d Mon Sep 17 00:00:00 2001 From: Yuanyuan Chen Date: Fri, 17 Oct 2025 21:00:57 +0800 Subject: [PATCH] Enable faiss-cpu on Windows (#41678) faiss-cpu is supported on Windows Signed-off-by: Yuanyuan Chen --- setup.py | 6 +----- 1 file changed, 1 insertion(+), 5 deletions(-) diff --git a/setup.py b/setup.py index 4b5141867da..319b08e5099 100644 --- a/setup.py +++ b/setup.py @@ -67,7 +67,6 @@ To create the package for pypi. 9. Copy the release notes from RELEASE.md to the tag in github once everything is looking hunky-dory. """ -import os import re import shutil from pathlib import Path @@ -261,10 +260,7 @@ extras["torch"] = deps_list("torch", "accelerate") extras["accelerate"] = deps_list("accelerate") extras["hf_xet"] = deps_list("hf_xet") -if os.name == "nt": # windows - extras["retrieval"] = deps_list("datasets") # faiss is not supported on windows -else: - extras["retrieval"] = deps_list("faiss-cpu", "datasets") +extras["retrieval"] = deps_list("faiss-cpu", "datasets") extras["tokenizers"] = deps_list("tokenizers") extras["ftfy"] = deps_list("ftfy")