Upload folder using huggingface_hub
Browse files
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()
|