mariusjabami commited on
Commit
3cb84b2
·
verified ·
1 Parent(s): f376c70

Update template

Browse files
Files changed (1) hide show
  1. template +17 -14
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
- {%- if (message['role'] == 'assistant') -%}
 
22
  {%- set role = "model" -%}
23
  {%- else -%}
24
  {%- set role = message['role'] -%}
25
  {%- endif -%}
26
- {{ '<start_of_turn>' + role + '
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
- {{ '<end_of_turn>
42
- ' }}
43
  {%- endfor -%}
 
44
  {%- if add_generation_prompt -%}
45
- {{'<start_of_turn>model
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 -%}