File size: 1,090 Bytes
66b6789
 
 
1
2
3
4
{
    "chat_template": "{%- if messages is string -%}\n    {{- messages -}}\n{%- else -%}\n    {%- for message in messages -%}\n        {%- if loop.first and messages[0]['role'] != 'system' -%}\n            {{- '<|im_start|>system\\nYou are a helpful assistant.<|im_end|>\\n' -}}\n        {%- endif -%}\n        {{- '<|im_start|>' + message['role'] + '\\n' -}}\n        {%- if message['content'] is string -%}\n            {{- message['content'] -}}\n        {%- elif message['content'] is iterable -%}\n            {%- for item in message['content'] -%}\n                {%- if item['type'] == 'image' -%}\n                    {{- '<image>\\n' -}}\n                {%- elif item['type'] == 'text' -%}\n                    {{- item['text'] -}}\n                {%- endif -%}\n            {%- endfor -%}\n        {%- else -%}\n            {{- raise_exception(\"Invalid content type\") -}}\n        {%- endif -%}\n        {{- '<|im_end|>' + '\\n' -}}\n    {%- endfor -%}\n    {%- if add_generation_prompt -%}\n        {{- '<|im_start|>assistant\\n' -}}\n    {%- endif -%}\n{%- endif -%}\n"
}