Update Mistral V1.yaml

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

View File

@ -18,7 +18,7 @@ instruction_template: |-
{{- ' [INST] ' + message['content'].rstrip() + ' [/INST]' }} {{- ' [INST] ' + message['content'].rstrip() + ' [/INST]' }}
{%- endif %} {%- endif %}
{%- elif message['role'] == 'assistant' %} {%- elif message['role'] == 'assistant' %}
{{- ' ' + message['content'].rstrip() + '</s>'}} {{- '' + message['content'] + '</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 %}