yujunjun/LightRAG
1
0
Fork 0
mirror of https://github.com/HKUDS/LightRAG.git synced 2025-08-12 18:50:47 +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,
)
import faiss # type: ignore
USE_GPU = os.getenv("FAISS_USE_GPU", "0") == "1"
FAISS_PACKAGE = "faiss-gpu" if USE_GPU else "faiss-cpu"
if not pm.is_installed(FAISS_PACKAGE):
pm.install(FAISS_PACKAGE)
import faiss # type: ignore
@final
@dataclass