cuierfei commited on
Commit
ce1559d
1 Parent(s): ecf72f7

Upload folder using huggingface_hub

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