Merge remote-tracking branch 'refs/remotes/origin/dev' into dev

This commit is contained in:
oobabooga 2024-01-09 16:28:14 -08:00
commit 038b4fc8af

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 -%}