Mirrors/text-generation-webui
1
0
Fork 0
mirror of https://github.com/oobabooga/text-generation-webui.git synced 2024-11-22 08:07:56 +01:00
Code Issues Packages Projects Releases Wiki Activity

Fix spaces in Mistral/Mixtral instruct prompt (#5214)

Browse Source
This commit is contained in:
Badis Ghoubali 2024-01-10 01:12:54 +01:00 committed by GitHub
parent bec4e0a1ce
commit c44836c4d7
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
instruction-templates/Mistral.yaml
View File

@ -4,13 +4,12 @@ instruction_template: |-
{{- message['content'] -}} {{- message['content'] -}}
{%- else -%} {%- else -%}
{%- if message['role'] == 'user' -%} {%- if message['role'] == 'user' -%}
{{-'[INST] ' + message['content'] + ' [/INST]'-}} {{-' [INST] ' + message['content'].rstrip() + ' [/INST] '-}}
{%- else -%} {%- else -%}
{{-'' + message['content'] + '</s> ' -}} {{-'' + message['content'] + '</s>' -}}
{%- endif -%} {%- endif -%}
{%- endif -%} {%- endif -%}
{%- endfor -%} {%- endfor -%}
{%- if add_generation_prompt -%} {%- if add_generation_prompt -%}
{{-''-}} {{-''-}}
{%- endif -%} {%- endif -%}