rootlocalghost commited on
Commit
3479e2e
·
verified ·
1 Parent(s): c18315d

clone chat_template.jinja

Browse files
Files changed (1) hide show
  1. chat_template.jinja +77 -0
chat_template.jinja ADDED
@@ -0,0 +1,77 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {%- if messages[0]['role'] == 'system' %}
2
+ {{- '<|im_start|>system\n' }}
3
+ {%- if messages[0]['content'] is string %}
4
+ {{- messages[0]['content'] }}
5
+ {%- else %}
6
+ {%- for item in messages[0]['content'] %}
7
+ {%- if item['type'] == 'text' %}
8
+ {{- item['text'] }}
9
+ {%- endif %}
10
+ {%- endfor %}
11
+ {%- endif %}
12
+ {{- '<|im_end|>\n' }}
13
+ {%- else %}
14
+ {{- '<|im_start|>system\n你是Qianfan-VL,由百度智能云研发的多模态大语言模型。<|im_end|>\n' }}
15
+ {%- endif %}
16
+ {%- set ns = namespace(found_last_user=false, last_query_index=messages|length - 1) %}
17
+ {%- for message in messages[::-1] %}
18
+ {%- set index = (messages|length - 1) - loop.index0 %}
19
+ {%- if not ns.found_last_user and message['role'] == 'user' %}
20
+ {%- set ns.found_last_user = true %}
21
+ {%- set ns.last_query_index = index %}
22
+ {%- endif %}
23
+ {%- endfor %}
24
+ {%- for message in messages %}
25
+ {%- if messages[0]['role'] != 'system' or not loop.first %}
26
+ {%- if message['role'] == 'user' or (message['role'] == 'system' and not loop.first) %}
27
+ {%- set append_think = (enable_thinking is defined and enable_thinking and message['role'] == 'user' and loop.index0 == ns.last_query_index) %}
28
+ {{- '<|im_start|>' + message['role'] + '\n' }}
29
+ {%- if message['content'] is string %}
30
+ {{- message['content'] }}
31
+ {%- else %}
32
+ {%- for item in message['content'] %}
33
+ {%- if item['type'] == 'image' %}
34
+ {{- '<image>\n' }}
35
+ {%- elif item['type'] == 'video' %}
36
+ {{- '<video>\n' }}
37
+ {%- elif item['type'] == 'text' %}
38
+ {{- item['text'] }}
39
+ {%- endif %}
40
+ {%- endfor %}
41
+ {%- endif %}
42
+ {%- if append_think %}
43
+ {{- '<think>' }}
44
+ {%- endif %}
45
+ {{- '<|im_end|>\n' }}
46
+ {%- elif message['role'] == 'assistant' %}
47
+ {%- if message['content'] is string %}
48
+ {%- set raw_content = message['content'] %}
49
+ {%- else %}
50
+ {%- set content_ns = namespace(raw='') %}
51
+ {%- for item in message['content'] %}
52
+ {%- if item['type'] == 'text' %}
53
+ {%- set content_ns.raw = content_ns.raw + item['text'] %}
54
+ {%- endif %}
55
+ {%- endfor %}
56
+ {%- set raw_content = content_ns.raw %}
57
+ {%- endif %}
58
+ {%- set content = raw_content %}
59
+ {%- set reasoning_content = '' %}
60
+ {%- if 'reasoning_content' in message and message['reasoning_content'] is not none %}
61
+ {%- set reasoning_content = message['reasoning_content'] %}
62
+ {%- elif '</think>' in raw_content %}
63
+ {%- set content = raw_content.split('</think>')[-1].lstrip('\n') %}
64
+ {%- set reasoning_content = raw_content.split('</think>')[0].rstrip('\n').split('<think>')[-1].lstrip('\n') %}
65
+ {%- endif %}
66
+ {%- if loop.index0 > ns.last_query_index and reasoning_content %}
67
+ {{- '<|im_start|>' + message['role'] + '\n<think>\n' + reasoning_content.strip('\n') + '\n</think>\n\n' + content.lstrip('\n') }}
68
+ {%- else %}
69
+ {{- '<|im_start|>' + message['role'] + '\n' + content }}
70
+ {%- endif %}
71
+ {{- '<|im_end|>\n' }}
72
+ {%- endif %}
73
+ {%- endif %}
74
+ {%- endfor %}
75
+ {%- if add_generation_prompt %}
76
+ {{- '<|im_start|>assistant\n' }}
77
+ {%- endif %}