File size: 529 Bytes
81975a7 087af17 81975a7 bdf4a39 81975a7 087af17 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 |
from transformers import LlavaNextConfig
class GraniteVisionEmbConfig(LlavaNextConfig):
model_type = "granitevisionemb"
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.base_image_feature_location = kwargs.get("base_image_feature_location", "last")
self.adapter_path = kwargs.get("adapter_path", None)
super().__init__(**kwargs)
|