default values for colbert

webis-de · Jul 18, 2024 · 3a5f9e7 · 3a5f9e7
1 parent f0fb1cf
commit 3a5f9e7
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/lightning_ir/models/col/config.py b/lightning_ir/models/col/config.py
@@ -12,10 +12,10 @@ def __init__(
         query_expansion: bool = True,
         attend_to_query_expanded_tokens: bool = False,
         query_mask_scoring_tokens: Sequence[str] | None = None,
-        doc_mask_scoring_tokens: (Sequence[str] | Literal["punctuation"] | None) = "punctuation",
+        doc_mask_scoring_tokens: Sequence[str] | Literal["punctuation"] | None = "punctuation",
         query_aggregation_function: Literal["sum", "mean", "max", "harmonic_mean"] = "sum",
-        normalize: bool = True,
-        add_marker_tokens: bool = True,
+        normalize: bool = False,
+        add_marker_tokens: bool = False,
         embedding_dim: int = 128,
         projection: Literal["linear", "linear_no_bias"] | None = "linear_no_bias",
         **kwargs,