diff --git a/lightning_ir/models/col/config.py b/lightning_ir/models/col/config.py index 58d8b1f..2b21e76 100644 --- 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,