Fix faisss import error
Browse files
lightrag/kg/faiss_impl.py
CHANGED
@@ -17,14 +17,13 @@ from .shared_storage import (
|
|
17 |
set_all_update_flags,
|
18 |
)
|
19 |
|
20 |
-
import faiss # type: ignore
|
21 |
-
|
22 |
USE_GPU = os.getenv("FAISS_USE_GPU", "0") == "1"
|
23 |
FAISS_PACKAGE = "faiss-gpu" if USE_GPU else "faiss-cpu"
|
24 |
-
|
25 |
if not pm.is_installed(FAISS_PACKAGE):
|
26 |
pm.install(FAISS_PACKAGE)
|
27 |
|
|
|
|
|
28 |
|
29 |
@final
|
30 |
@dataclass
|
|
|
17 |
set_all_update_flags,
|
18 |
)
|
19 |
|
|
|
|
|
20 |
USE_GPU = os.getenv("FAISS_USE_GPU", "0") == "1"
|
21 |
FAISS_PACKAGE = "faiss-gpu" if USE_GPU else "faiss-cpu"
|
|
|
22 |
if not pm.is_installed(FAISS_PACKAGE):
|
23 |
pm.install(FAISS_PACKAGE)
|
24 |
|
25 |
+
import faiss # type: ignore
|
26 |
+
|
27 |
|
28 |
@final
|
29 |
@dataclass
|