diff --git a/instruction-templates/Mistral V1.yaml b/instruction-templates/Mistral V1.yaml index 46194b66..950be5c7 100644 --- a/instruction-templates/Mistral V1.yaml +++ b/instruction-templates/Mistral V1.yaml @@ -18,7 +18,7 @@ instruction_template: |- {{- ' [INST] ' + message['content'].rstrip() + ' [/INST]' }} {%- endif %} {%- elif message['role'] == 'assistant' %} - {{- ' ' + message['content'].rstrip() + ''}} + {{- '' + message['content'] + ''}} {%- else %} {{- raise_exception('Only user and assistant roles are supported, with the exception of an initial optional system message!') }} {%- endif %}