Mizukiluke commited on
Commit
8ca1d87
1 Parent(s): 7c86757

Update processing_mplugowl3.py

Browse files
Files changed (1) hide show
  1. 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