yujunjun/LightRAG
1
0
Fork 0
mirror of https://github.com/HKUDS/LightRAG.git synced 2025-08-13 03:00:48 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix faisss import error

Browse Source
This commit is contained in:
yangdx 2025-07-04 19:12:23 +08:00
parent 531502677e
commit 810676d54c
1 changed files with 2 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
lightrag/kg/faiss_impl.py
View File

@ -17,14 +17,13 @@ from .shared_storage import (
set_all_update_flags, set_all_update_flags,
) )
import faiss # type: ignore
USE_GPU = os.getenv("FAISS_USE_GPU", "0") == "1" USE_GPU = os.getenv("FAISS_USE_GPU", "0") == "1"
FAISS_PACKAGE = "faiss-gpu" if USE_GPU else "faiss-cpu" FAISS_PACKAGE = "faiss-gpu" if USE_GPU else "faiss-cpu"
if not pm.is_installed(FAISS_PACKAGE): if not pm.is_installed(FAISS_PACKAGE):
pm.install(FAISS_PACKAGE) pm.install(FAISS_PACKAGE)
import faiss # type: ignore
@final @final
@dataclass @dataclass