Upload colgranitevision_config.py

colgranitevision_config.py

@@ -0,0 +1,12 @@
+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)