Update Mistral V1.yaml

This commit is contained in:
pandora 2024-09-21 18:54:48 +02:00 committed by GitHub
parent 015ad8505b
commit c6047508b7
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -13,12 +13,12 @@ instruction_template: |-
{%- endif %} {%- endif %}
{%- if message['role'] == 'user' %} {%- if message['role'] == 'user' %}
{%- if loop.first and system_message is defined %} {%- if loop.first and system_message is defined %}
{{- ' [INST] ' + system_message + '\\n\\n' + message['content'] + ' [/INST]' }} {{- ' [INST] ' + system_message + '\\n\\n' + message['content'].rstrip() + ' [/INST]' }}
{%- else %} {%- else %}
{{- ' [INST] ' + message['content'] + ' [/INST]' }} {{- ' [INST] ' + message['content'].rstrip() + ' [/INST]' }}
{%- endif %} {%- endif %}
{%- elif message['role'] == 'assistant' %} {%- elif message['role'] == 'assistant' %}
{{- ' ' + message['content'] + '</s>'}} {{- ' ' + message['content'].rstrip() + '</s>'}}
{%- else %} {%- else %}
{{- raise_exception('Only user and assistant roles are supported, with the exception of an initial optional system message!') }} {{- raise_exception('Only user and assistant roles are supported, with the exception of an initial optional system message!') }}
{%- endif %} {%- endif %}