File size: 441 Bytes
92c7321 91a3777 c6bc767 2c87b68 91a3777 | 1 2 3 4 5 6 7 8 9 10 11 12 13 | from models.lm.config import LMConfig
class VLMConfig(LMConfig):
model_type = "omni-v"
def __init__(self, image_special_token='<|image_pad|>', image_ids=[12], **kwargs):
self.image_special_token = image_special_token
self.image_ids = image_ids
self.image_hidden_size = kwargs.get("image_hidden_size", 768)
self.image_token_len = kwargs.get("image_token_len", 64)
super().__init__(**kwargs)
|