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