Add arguments to MiniCPMVConfig for vLLM support
Browse files- configuration_minicpm.py +10 -0
configuration_minicpm.py
CHANGED
@@ -209,6 +209,11 @@ class MiniCPMVConfig(MiniCPMConfig):
|
|
209 |
patch_size=14,
|
210 |
max_slice_nums=9,
|
211 |
scale_resolution=448,
|
|
|
|
|
|
|
|
|
|
|
212 |
**kwargs,
|
213 |
):
|
214 |
self.vision_encoder = vision_encoder
|
@@ -219,4 +224,9 @@ class MiniCPMVConfig(MiniCPMConfig):
|
|
219 |
self.patch_size = patch_size
|
220 |
self.max_slice_nums = max_slice_nums
|
221 |
self.scale_resolution = scale_resolution
|
|
|
|
|
|
|
|
|
|
|
222 |
super().__init__(**kwargs)
|
|
|
209 |
patch_size=14,
|
210 |
max_slice_nums=9,
|
211 |
scale_resolution=448,
|
212 |
+
im_start_token_id=101,
|
213 |
+
im_end_token_id=102,
|
214 |
+
slice_start_token_id=111,
|
215 |
+
slice_end_token_id=112,
|
216 |
+
unk_token_id=0,
|
217 |
**kwargs,
|
218 |
):
|
219 |
self.vision_encoder = vision_encoder
|
|
|
224 |
self.patch_size = patch_size
|
225 |
self.max_slice_nums = max_slice_nums
|
226 |
self.scale_resolution = scale_resolution
|
227 |
+
self.im_start_token_id = im_start_token_id
|
228 |
+
self.im_end_token_id = im_end_token_id
|
229 |
+
self.slice_start_token_id = slice_start_token_id
|
230 |
+
self.slice_end_token_id = slice_end_token_id
|
231 |
+
self.unk_token_id = unk_token_id
|
232 |
super().__init__(**kwargs)
|