File size: 434 Bytes
81975a7 |
1 2 3 4 5 6 7 8 9 10 11 12 13 |
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)
|