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()
|