sharp1112 commited on
Commit
cd3d500
·
verified ·
1 Parent(s): 06b404f

Upload chat_template.jinja with huggingface_hub

Browse files
Files changed (1) hide show
  1. chat_template.jinja +90 -0
chat_template.jinja ADDED
@@ -0,0 +1,90 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {%- if tools %}
2
+ {{- '<|im_start|>system\n' }}
3
+ {%- if messages[0].role == 'system' %}
4
+ {{- messages[0].content + '\n\n' }}
5
+ {%- endif %}
6
+ {{- "# Tools\n\nYou may call one or more functions to assist with the user query.\n\nYou are provided with function signatures within <tools></tools> XML tags:\n<tools>" }}
7
+ {%- for tool in tools %}
8
+ {{- "\n" }}
9
+ {{- tool | tojson }}
10
+ {%- endfor %}
11
+ {{- "\n</tools>\n\nFor each function call, return a json object with function name and arguments within <tool_call></tool_call> XML tags:\n<tool_call>\n{\"name\": <function-name>, \"arguments\": <args-json-object>}\n</tool_call><|im_end|>\n" }}
12
+ {%- else %}
13
+ {%- if messages[0].role == 'system' %}
14
+ {{- '<|im_start|>system\n' + messages[0].content + '<|im_end|>\n' }}
15
+ {%- endif %}
16
+ {%- endif %}
17
+ {%- for message in messages %}
18
+ {%- if message.content is string %}
19
+ {%- set content = message.content %}
20
+ {%- else %}
21
+ {%- set content = '' %}
22
+ {%- endif %}
23
+ {%- if (message.role == "user") or (message.role == "system" and not loop.first) %}
24
+ {{- '<|im_start|>' + message.role + '\n' + content + '<|im_end|>' + '\n' }}
25
+ {%- elif message.role == "assistant" %}
26
+ {{- '<|im_start|>' + message.role + '\n' }}
27
+ {%- set has_loss = (message.loss is defined and message.loss) %}
28
+ {%- if has_loss %}
29
+ {% generation %}
30
+ {{- content }}
31
+ {%- if message.tool_calls %}
32
+ {%- for tool_call in message.tool_calls %}
33
+ {%- if (loop.first and content) or (not loop.first) %}
34
+ {{- '\n' }}
35
+ {%- endif %}
36
+ {%- if tool_call.function %}
37
+ {%- set tool_call = tool_call.function %}
38
+ {%- endif %}
39
+ {{- '<tool_call>\n{"name": "' }}
40
+ {{- tool_call.name }}
41
+ {{- '", "arguments": ' }}
42
+ {%- if tool_call.arguments is string %}
43
+ {{- tool_call.arguments }}
44
+ {%- else %}
45
+ {{- tool_call.arguments | tojson }}
46
+ {%- endif %}
47
+ {{- '}\n</tool_call>' }}
48
+ {%- endfor %}
49
+ {%- endif %}
50
+ {{- '<|im_end|>' }}
51
+ {%- endgeneration %}
52
+ {{- '\n' }}
53
+ {%- else %}
54
+ {{- content }}
55
+ {%- if message.tool_calls %}
56
+ {%- for tool_call in message.tool_calls %}
57
+ {%- if (loop.first and content) or (not loop.first) %}
58
+ {{- '\n' }}
59
+ {%- endif %}
60
+ {%- if tool_call.function %}
61
+ {%- set tool_call = tool_call.function %}
62
+ {%- endif %}
63
+ {{- '<tool_call>\n{"name": "' }}
64
+ {{- tool_call.name }}
65
+ {{- '", "arguments": ' }}
66
+ {%- if tool_call.arguments is string %}
67
+ {{- tool_call.arguments }}
68
+ {%- else %}
69
+ {{- tool_call.arguments | tojson }}
70
+ {%- endif %}
71
+ {{- '}\n</tool_call>' }}
72
+ {%- endfor %}
73
+ {%- endif %}
74
+ {{- '<|im_end|>\n' }}
75
+ {%- endif %}
76
+ {%- elif message.role == "tool" %}
77
+ {%- if loop.first or (messages[loop.index0 - 1].role != "tool") %}
78
+ {{- '<|im_start|>user' }}
79
+ {%- endif %}
80
+ {{- '\n<tool_response>\n' }}
81
+ {{- content }}
82
+ {{- '\n</tool_response>' }}
83
+ {%- if loop.last or (messages[loop.index0 + 1].role != "tool") %}
84
+ {{- '<|im_end|>\n' }}
85
+ {%- endif %}
86
+ {%- endif %}
87
+ {%- endfor %}
88
+ {%- if add_generation_prompt %}
89
+ {{- '<|im_start|>assistant\n' }}
90
+ {%- endif %}