Mizukiluke
commited on
Commit
•
8ca1d87
1
Parent(s):
7c86757
Update processing_mplugowl3.py
Browse files- processing_mplugowl3.py +2 -0
processing_mplugowl3.py
CHANGED
@@ -232,6 +232,8 @@ class mPLUGOwl3Processor(ProcessorMixin):
|
|
232 |
text += self.image_processor.cut_prompt_template(img_token='<|image|>', h=cut_shape[image_token_ptr][0], w=cut_shape[image_token_ptr][1])
|
233 |
text += next_text
|
234 |
image_token_ptr += 1
|
|
|
|
|
235 |
message['content'] = text
|
236 |
|
237 |
|
|
|
232 |
text += self.image_processor.cut_prompt_template(img_token='<|image|>', h=cut_shape[image_token_ptr][0], w=cut_shape[image_token_ptr][1])
|
233 |
text += next_text
|
234 |
image_token_ptr += 1
|
235 |
+
if self.image_processor.add_global:
|
236 |
+
image_token_ptr += 1 # skip the global image which has been include by cut_prompt_template
|
237 |
message['content'] = text
|
238 |
|
239 |
|