diff --git a/lightning_ir/retrieve/faiss/faiss_searcher.py b/lightning_ir/retrieve/faiss/faiss_searcher.py index f5a4e14..8c7ebd8 100644 --- a/lightning_ir/retrieve/faiss/faiss_searcher.py +++ b/lightning_ir/retrieve/faiss/faiss_searcher.py @@ -19,13 +19,12 @@ def __init__( search_config: FaissSearchConfig, module: BiEncoderModule, use_gpu: bool = False, - use_gpu_index: bool = False, ) -> None: import faiss self.search_config: FaissSearchConfig self.index = faiss.read_index(str(Path(index_dir) / "index.faiss")) - if use_gpu and use_gpu_index and hasattr(faiss, "index_cpu_to_all_gpus"): + if use_gpu and hasattr(faiss, "index_cpu_to_all_gpus"): self.index = faiss.index_cpu_to_all_gpus(self.index) ivf_index = None try: