gzdaniel commited on
Commit
6a0cb6d
·
1 Parent(s): 483bc22

Fix faisss import error

Browse files
Files changed (1) hide show
  1. lightrag/kg/faiss_impl.py +2 -3
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