File size: 665 Bytes
aac158a |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 |
from transformers import PretrainedConfig
from typing import List
class InfMLLMChatConfig(PretrainedConfig):
def __init__(
self,
image_size=448,
vision_tower="eva_clip_g",
mm_projector_type="pooler",
pool_out_size="32+16+8",
mm_hidden_size=1408,
hidden_size=4096,
**kwargs
):
self.image_size = image_size
self.vision_tower = vision_tower
self.mm_projector_type = mm_projector_type
self.pool_out_size = pool_out_size
self.mm_hidden_size = mm_hidden_size
self.hidden_size = hidden_size
super().__init__(**kwargs)
|