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 conversion from old template format to jinja2

Browse Source
This commit is contained in:
oobabooga 2023-12-19 13:16:52 -08:00
parent de138b8ba6
commit f4ae0075e8
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
modules/chat.py
View File

@ -768,13 +768,13 @@ def delete_character(name, instruct=False):
def jinja_template_from_old_format(params, verbose=False): def jinja_template_from_old_format(params, verbose=False):
MASTER_TEMPLATE = """ MASTER_TEMPLATE = """
{%- set found_item = false -%} {%- set ns = namespace(found=false) -%}
{%- for message in messages -%} {%- for message in messages -%}
{%- if message['role'] == 'system' -%} {%- if message['role'] == 'system' -%}
{%- set found_item = true -%} {%- set ns.found = true -%}
{%- endif -%} {%- endif -%}
{%- endfor -%} {%- endfor -%}
{%- if not found_item -%} {%- if not ns.found -%}
{{- '<|PRE-SYSTEM|>' + '<|SYSTEM-MESSAGE|>' + '<|POST-SYSTEM|>' -}} {{- '<|PRE-SYSTEM|>' + '<|SYSTEM-MESSAGE|>' + '<|POST-SYSTEM|>' -}}
{%- endif %} {%- endif %}
{%- for message in messages %} {%- for message in messages %}