Update template
Browse files
template
CHANGED
|
@@ -1,30 +1,33 @@
|
|
| 1 |
{{ bos_token }}
|
|
|
|
|
|
|
|
|
|
|
|
|
| 2 |
{%- if messages[0]['role'] == 'system' -%}
|
| 3 |
{%- if messages[0]['content'] is string -%}
|
| 4 |
-
{%- set first_user_prefix = messages[0]['content'] +
|
| 5 |
-
|
| 6 |
-
' -%}
|
| 7 |
{%- else -%}
|
| 8 |
-
{%- set first_user_prefix = messages[0]['content'][0]['text'] +
|
| 9 |
-
|
| 10 |
-
' -%}
|
| 11 |
{%- endif -%}
|
| 12 |
{%- set loop_messages = messages[1:] -%}
|
| 13 |
{%- else -%}
|
| 14 |
-
{%- set first_user_prefix = "" -%}
|
| 15 |
{%- set loop_messages = messages -%}
|
| 16 |
{%- endif -%}
|
|
|
|
| 17 |
{%- for message in loop_messages -%}
|
| 18 |
{%- if (message['role'] == 'user') != (loop.index0 % 2 == 0) -%}
|
| 19 |
{{ raise_exception("Conversation roles must alternate user/assistant/user/assistant/...") }}
|
| 20 |
{%- endif -%}
|
| 21 |
-
|
|
|
|
| 22 |
{%- set role = "model" -%}
|
| 23 |
{%- else -%}
|
| 24 |
{%- set role = message['role'] -%}
|
| 25 |
{%- endif -%}
|
| 26 |
-
|
| 27 |
-
' + (first_user_prefix if loop.first else "") }}
|
|
|
|
| 28 |
{%- if message['content'] is string -%}
|
| 29 |
{{ message['content'] | trim }}
|
| 30 |
{%- elif message['content'] is iterable -%}
|
|
@@ -38,10 +41,10 @@
|
|
| 38 |
{%- elif message['content'] is defined -%}
|
| 39 |
{{ raise_exception("Invalid content type") }}
|
| 40 |
{%- endif -%}
|
| 41 |
-
|
| 42 |
-
' }}
|
| 43 |
{%- endfor -%}
|
|
|
|
| 44 |
{%- if add_generation_prompt -%}
|
| 45 |
-
|
| 46 |
-
'}}
|
| 47 |
{%- endif -%}
|
|
|
|
| 1 |
{{ bos_token }}
|
| 2 |
+
{%- set default_system_message =
|
| 3 |
+
"Você é o modelo Link, treinado pela LambdaXi Corp. Criador: Marius Jabami."
|
| 4 |
+
-%}
|
| 5 |
+
|
| 6 |
{%- if messages[0]['role'] == 'system' -%}
|
| 7 |
{%- if messages[0]['content'] is string -%}
|
| 8 |
+
{%- set first_user_prefix = default_system_message + "\n\n" + messages[0]['content'] + "\n\n" -%}
|
|
|
|
|
|
|
| 9 |
{%- else -%}
|
| 10 |
+
{%- set first_user_prefix = default_system_message + "\n\n" + messages[0]['content'][0]['text'] + "\n\n" -%}
|
|
|
|
|
|
|
| 11 |
{%- endif -%}
|
| 12 |
{%- set loop_messages = messages[1:] -%}
|
| 13 |
{%- else -%}
|
| 14 |
+
{%- set first_user_prefix = default_system_message + "\n\n" -%}
|
| 15 |
{%- set loop_messages = messages -%}
|
| 16 |
{%- endif -%}
|
| 17 |
+
|
| 18 |
{%- for message in loop_messages -%}
|
| 19 |
{%- if (message['role'] == 'user') != (loop.index0 % 2 == 0) -%}
|
| 20 |
{{ raise_exception("Conversation roles must alternate user/assistant/user/assistant/...") }}
|
| 21 |
{%- endif -%}
|
| 22 |
+
|
| 23 |
+
{%- if message['role'] == 'assistant' -%}
|
| 24 |
{%- set role = "model" -%}
|
| 25 |
{%- else -%}
|
| 26 |
{%- set role = message['role'] -%}
|
| 27 |
{%- endif -%}
|
| 28 |
+
|
| 29 |
+
{{ '<start_of_turn>' + role + '\n' + (first_user_prefix if loop.first else "") }}
|
| 30 |
+
|
| 31 |
{%- if message['content'] is string -%}
|
| 32 |
{{ message['content'] | trim }}
|
| 33 |
{%- elif message['content'] is iterable -%}
|
|
|
|
| 41 |
{%- elif message['content'] is defined -%}
|
| 42 |
{{ raise_exception("Invalid content type") }}
|
| 43 |
{%- endif -%}
|
| 44 |
+
|
| 45 |
+
{{ '<end_of_turn>\n' }}
|
| 46 |
{%- endfor -%}
|
| 47 |
+
|
| 48 |
{%- if add_generation_prompt -%}
|
| 49 |
+
{{ '<start_of_turn>model\n' }}
|
|
|
|
| 50 |
{%- endif -%}
|