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