Update README.md

README.md

@@ -212,7 +212,7 @@ Output:
 Retrievers and Readers can be used separately.
 In the case of retriever-only ReLiK, the output will contain the candidates for the input text.
 
-Reader-only example:
+Retriever-only example:
 
 ```python
 from relik import Relik
@@ -246,7 +246,7 @@ Output:
       ),
     )
 
-Retriever-only example:
+Reader-only example:
 
 ```python
 from relik import Relik

relik/common/utils.py

@@ -339,9 +339,10 @@ def download_from_hf(
             downloaded_paths.append(downloaded_path)
         except OSError:
             if ignore_failure:
-                logger.warn(
-                    f"Couldn't download {filename} from {path_or_repo_id}, ignoring"
-                )
+                # logger.warn(
+                #     f"Couldn't download {filename} from {path_or_repo_id}, ignoring"
+                # )
+                pass
             else:
                 raise
 

relik/retriever/indexers/base.py

@@ -467,13 +467,15 @@ def from_pretrained(
             )
 
         config = OmegaConf.load(config_path)
-        # add the actual cls class to the config in place of the _target_ if cls is not BaseDocumentIndex
 
+        # add the actual cls class to the config in place of the _target_ if cls is not BaseDocumentIndex
         target = config.get("_target_", None)
+        use_faiss = kwargs.get("use_faiss", False) or "FaissDocumentIndex" in target
+
         if cls.__name__ != "BaseDocumentIndex" and target is None:
             kwargs["_target_"] = f"{cls.__module__}.{cls.__name__}"
-
-        use_faiss = kwargs.get("use_faiss", False) or "FaissDocumentIndex" in target
+        if use_faiss and "FaissDocumentIndex" not in target:
+            kwargs["_target_"] = "relik.retriever.indexers.faissindex.FaissDocumentIndex"
 
         kwargs["device"] = device
         kwargs["precision"] = precision

relik/retriever/indexers/faissindex.py

@@ -179,8 +179,8 @@ def _build_faiss_index(
                 embeddings.cpu() if isinstance(embeddings, torch.Tensor) else embeddings
             )
 
-        # convert to float32 if embeddings is a torch.Tensor and is float16
-        if isinstance(embeddings, torch.Tensor) and embeddings.dtype == torch.float16:
+        # convert to float32 if embeddings is a torch.Tensor and not already float32
+        if isinstance(embeddings, torch.Tensor) and embeddings.dtype != torch.float32:
             embeddings = embeddings.float()
 
         # logger.info("Training the index.")

relik/retriever/indexers/inmemory.py

@@ -41,7 +41,7 @@ def __init__(
         separator: str | None = None,
         name_or_path: str | os.PathLike | None = None,
         device: str = "cpu",
-        precision: str | int | torch.dtype = 32,
+        precision: str | int | torch.dtype | None = 32,
         *args,
         **kwargs,
     ) -> None:

relik/version.py

@@ -4,7 +4,7 @@
 _MINOR = "0"
 # On main and in a nightly release the patch should be one ahead of the last
 # released build.
-_PATCH = "5"
+_PATCH = "6"
 # This is mainly for nightly builds which have the suffix ".dev$DATE". See
 # https://semver.org/#is-v123-a-semantic-version for the semantics.
 _SUFFIX = os.environ.get("RELIK_VERSION_SUFFIX", "")