frozenleaves/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-10-20 17:13:56 +08:00
Code Issues Packages Projects Releases Wiki Activity

Enable faiss-cpu on Windows (#41678)

Browse Source 
faiss-cpu is supported on Windows

Signed-off-by: Yuanyuan Chen <cyyever@outlook.com>
This commit is contained in:
Yuanyuan Chen
2025-10-17 21:00:57 +08:00
committed by GitHub
parent 10de06dace
commit c01ceffeb4
1 changed files with 1 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
setup.py
View File

@ -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. 9. Copy the release notes from RELEASE.md to the tag in github once everything is looking hunky-dory.
""" """
import os
import re import re
import shutil import shutil
from pathlib import Path from pathlib import Path
@ -261,10 +260,7 @@ extras["torch"] = deps_list("torch", "accelerate")
extras["accelerate"] = deps_list("accelerate") extras["accelerate"] = deps_list("accelerate")
extras["hf_xet"] = deps_list("hf_xet") extras["hf_xet"] = deps_list("hf_xet")
if os.name == "nt": # windows extras["retrieval"] = deps_list("faiss-cpu", "datasets")
extras["retrieval"] = deps_list("datasets") # faiss is not supported on windows
else:
extras["retrieval"] = deps_list("faiss-cpu", "datasets")
extras["tokenizers"] = deps_list("tokenizers") extras["tokenizers"] = deps_list("tokenizers")
extras["ftfy"] = deps_list("ftfy") extras["ftfy"] = deps_list("ftfy")