Add skip_metadata parameter to RelikServer

SapienzaNLP · Aug 6, 2024 · 28e441b · 28e441b
1 parent ee6f03f
commit 28e441b
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/relik/inference/serve/backend/fastapi_be.py b/relik/inference/serve/backend/fastapi_be.py
@@ -52,6 +52,7 @@ def __init__(
         document_index_precision: str | int | torch.dtype | None = None,
         reader_precision: str | int | torch.dtype | None = None,
         annotation_type: str = "char",
+        skip_metadata: bool = False,
         **kwargs,
     ):
         num_threads = os.getenv("TORCH_NUM_THREADS", psutil.cpu_count(logical=False))
@@ -95,7 +96,7 @@ def __init__(
             retriever_precision=self.retriever_precision,
             document_index_precision=self.document_index_precision,
             reader_precision=self.reader_precision,
-            skip_metadata=True,
+            skip_metadata=skip_metadata,
         )
 
         self.router = APIRouter()