17 lines
397 B
Python
17 lines
397 B
Python
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)
|