apunkt/doc-to-lora
1
0
Fork 0
mirror of https://github.com/SakanaAI/doc-to-lora.git synced 2026-07-23 17:01:04 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix template for gemm3-1b

Browse source
This commit is contained in:
51616 2025-06-25 23:20:53 +09:00
parent fce138ff04
commit 13a062fad4
1 changed files with 7 additions and 28 deletions
Download patch file Download diff file Expand all files Collapse all files

35
chat_templates/google/gemma-3-1b-it.jinja
View file

@ -1,13 +1,9 @@
{{ bos_token }}
{%- if messages[0]['role'] == 'system' -%}
{%- if messages[0]['content'] is string -%}
{%- set first_user_prefix = messages[0]['content'] + '
' -%}
{%- set first_user_prefix = messages[0]['content'] + '\n\n' -%}
{%- else -%}
{%- set first_user_prefix = messages[0]['content'][0]['text'] + '
' -%}
{%- set first_user_prefix = messages[0]['content'][0]['text'] + '\n\n' -%}
{%- endif -%}
{%- set loop_messages = messages[1:] -%}
{%- else -%}
@ -23,29 +19,12 @@
{%- else -%}
{%- set role = message['role'] -%}
{%- endif -%}
{{ '<start_of_turn>' + role + '
' + (first_user_prefix if loop.first else "") }}
{%- if message['content'] is string -%}
{%- if message['role'] == 'assistant' -%}
{% generation %}{{ message['content'] | trim + '<end_of_turn>
' }}{% endgeneration %}
{%- else -%}
{{ message['content'] | trim + '<end_of_turn>
' }}
{%- endif -%}
{# {%- elif message['content'] is iterable -%}
{%- for item in message['content'] -%}
{%- if item['type'] == 'image' -%}
{{ '<start_of_image>' }}
{%- elif item['type'] == 'text' -%}
{{ item['text'] | trim }}
{%- endif -%}
{%- endfor -%} #}
{%- else -%}
{{ raise_exception("Invalid content type") }}
{%- if (message['role'] == 'user') -%}
{{ '<start_of_turn>' + role + '\n' + message['content'] | trim + '<end_of_turn>\n' }}
{%- elif (message['role'] == 'assistant') -%}
{{ '<start_of_turn>' + role + '\n' }}{% generation %}{{ message['content'] | trim + '<end_of_turn>\n' }}{% endgeneration %}
{%- endif -%}
{%- endfor -%}
{%- if add_generation_prompt -%}
{{'<start_of_turn>model
'}}
{{'<start_of_turn>model\n'}}
{%- endif -%}