cuierfei commited on
Commit
1da47c8
1 Parent(s): 3a3da7e

Upload folder using huggingface_hub

Browse files
Files changed (1) hide show
  1. modeling_internvl_chat.py +1 -0
modeling_internvl_chat.py CHANGED
@@ -216,6 +216,7 @@ class InternVLChatModel(PreTrainedModel):
216
  if pixel_values is not None and '<image>' not in question:
217
  question = '<image>\n' + question
218
  template = get_conv_template(self.template)
 
219
  template.append_message(template.roles[0], question)
220
  template.append_message(template.roles[1], None)
221
  query = template.get_prompt()
 
216
  if pixel_values is not None and '<image>' not in question:
217
  question = '<image>\n' + question
218
  template = get_conv_template(self.template)
219
+ template.system_message = self.system_message
220
  template.append_message(template.roles[0], question)
221
  template.append_message(template.roles[1], None)
222
  query = template.get_prompt()