nikita-savelyev-intel
commited on
Commit
•
1dab341
1
Parent(s):
026a06c
Create chat_template.json
Browse filesCopied from https://huggingface.co/llava-hf/llava-v1.6-mistral-7b-hf/blob/main/chat_template.json . It is needed for loading AutoProcessor from model_id. AutoProcessor in turn is required for collecting calibration data for quantization.
- chat_template.json +3 -0
chat_template.json
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"chat_template": "{% for message in messages %}{% if message['role'] == 'system' %}{{ '<<SYS>>\n' + message['content'][0]['text'] + '\n<</SYS>>\n\n' }}{% elif message['role'] == 'user' %}{{ '[INST] ' }}{# Render all images first #}{% for content in message['content'] | selectattr('type', 'equalto', 'image') %}{{ '<image>\n' }}{% endfor %}{# Render all text next #}{% for content in message['content'] | selectattr('type', 'equalto', 'text') %}{{ content['text'] }}{% endfor %}{{' [/INST]' }}{% elif message['role'] == 'assistant' %}{{ ' ' + message['content'][0]['text'] + '<\\s> '}}{% else %}{{ raise_exception('Only user and assistant roles are supported!') }}{% endif %}{% endfor %}"
|
3 |
+
}
|