File size: 794 Bytes
13f969b | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 | """
Configuration for ColVec1 retrieval model.
"""
from transformers.configuration_utils import PretrainedConfig
class ColVec1Config(PretrainedConfig):
"""Configuration for the ColVec1 retrieval wrapper."""
model_type = "colvec1"
def __init__(
self,
embed_dim: int = 128,
text_hidden_size: int = 2560,
padding_side: str = "left",
initializer_range: float = 0.02,
base_model_name_or_path: str = None,
**kwargs,
):
super().__init__(**kwargs)
self.embed_dim = embed_dim
self.text_hidden_size = text_hidden_size
self.padding_side = padding_side
self.initializer_range = initializer_range
self.base_model_name_or_path = base_model_name_or_path
__all__ = ["ColVec1Config"]
|