From 810676d54cae9b9df8322ce4e733a613cf2ec78c Mon Sep 17 00:00:00 2001 From: yangdx Date: Fri, 4 Jul 2025 19:12:23 +0800 Subject: [PATCH] Fix faisss import error --- lightrag/kg/faiss_impl.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/lightrag/kg/faiss_impl.py b/lightrag/kg/faiss_impl.py index f2afde2e..cb19497a 100644 --- a/lightrag/kg/faiss_impl.py +++ b/lightrag/kg/faiss_impl.py @@ -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