File size: 413 Bytes
d4c27fa |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 |
from .LMConfig import LMConfig
from typing import List
class VLMConfig(LMConfig):
model_type = "minimind-v"
def __init__(
self,
image_special_token: str = '@' * 196,
image_ids: List = [34] * 196,
**kwargs,
):
self.image_special_token = image_special_token
self.image_ids = image_ids
super().__init__(**kwargs)
|