Update config.py
Browse files
config.py
CHANGED
@@ -19,6 +19,8 @@ class LinearMappingConfig:
|
|
19 |
image_resize: int = 224
|
20 |
add_image_token: bool = True
|
21 |
freeze_ln: bool = False
|
|
|
|
|
22 |
|
23 |
def __post_init__(self):
|
24 |
self.prefix_length = PREFIX_MAP[self.image_model]
|
|
|
19 |
image_resize: int = 224
|
20 |
add_image_token: bool = True
|
21 |
freeze_ln: bool = False
|
22 |
+
image_from_pretrained: bool = True
|
23 |
+
text_from_pretrained: bool = True
|
24 |
|
25 |
def __post_init__(self):
|
26 |
self.prefix_length = PREFIX_MAP[self.image_model]
|