czczup commited on
Commit
53ba6de
1 Parent(s): ef6b153

Upload folder using huggingface_hub

Browse files
Files changed (1) hide show
  1. modeling_internvl_chat.py +3 -0
modeling_internvl_chat.py CHANGED
@@ -76,6 +76,8 @@ class InternVLChatModel(PreTrainedModel):
76
  )
77
 
78
  self.img_context_token_id = None
 
 
79
 
80
  def forward(
81
  self,
@@ -255,6 +257,7 @@ class InternVLChatModel(PreTrainedModel):
255
  self.img_context_token_id = img_context_token_id
256
 
257
  template = get_conv_template(self.template)
 
258
  eos_token_id = tokenizer.convert_tokens_to_ids(template.sep)
259
 
260
  history = [] if history is None else history
 
76
  )
77
 
78
  self.img_context_token_id = None
79
+ self.conv_template = get_conv_template(self.template)
80
+ self.system_message = self.conv_template.system_message
81
 
82
  def forward(
83
  self,
 
257
  self.img_context_token_id = img_context_token_id
258
 
259
  template = get_conv_template(self.template)
260
+ template.system_message = self.system_message
261
  eos_token_id = tokenizer.convert_tokens_to_ids(template.sep)
262
 
263
  history = [] if history is None else history