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