From fa1019e8fed7d94ff9f01c4baf6902a289b1bcfc Mon Sep 17 00:00:00 2001 From: fschuh Date: Fri, 16 Feb 2024 19:40:51 -0500 Subject: [PATCH] Removed extra spaces from Mistral instruction template that were causing Mistral to misbehave (#5517) --- instruction-templates/Mistral.yaml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/instruction-templates/Mistral.yaml b/instruction-templates/Mistral.yaml index 113c238b..02e5b20d 100644 --- a/instruction-templates/Mistral.yaml +++ b/instruction-templates/Mistral.yaml @@ -4,7 +4,7 @@ instruction_template: |- {{- message['content'] -}} {%- else -%} {%- if message['role'] == 'user' -%} - {{-' [INST] ' + message['content'].rstrip() + ' [/INST] '-}} + {{-'[INST] ' + message['content'].rstrip() + ' [/INST]'-}} {%- else -%} {{-'' + message['content'] + '' -}} {%- endif -%}