from transformers import LlavaNextConfig


class ColGraniteVisionConfig(LlavaNextConfig):
    model_type = "colgranitevision"

    def __init__(self, **kwargs):
        self.base_model = kwargs.get("base_model", None)
        self.emb_dim_query = kwargs.get("emb_dim_query", 128)
        self.emb_dim_doc = kwargs.get("emb_dim_doc", 128)
        self.adapter_path = kwargs.get("adapter_path", None)
        super().__init__(**kwargs)