zheqiushui commited on
Commit
50cd6d5
·
verified ·
1 Parent(s): 5fabff1

Upload 38 files

Browse files
Files changed (39) hide show
  1. .gitattributes +2 -0
  2. main_tp_api_x86 +3 -0
  3. main_tp_x86 +3 -0
  4. qwen2.5_7b_tp/model.embed_tokens.weight.bfloat16.bin +3 -0
  5. qwen2.5_7b_tp/qwen2_p128_l0_together.tar +3 -0
  6. qwen2.5_7b_tp/qwen2_p128_l10_together.tar +3 -0
  7. qwen2.5_7b_tp/qwen2_p128_l11_together.tar +3 -0
  8. qwen2.5_7b_tp/qwen2_p128_l12_together.tar +3 -0
  9. qwen2.5_7b_tp/qwen2_p128_l13_together.tar +3 -0
  10. qwen2.5_7b_tp/qwen2_p128_l14_together.tar +3 -0
  11. qwen2.5_7b_tp/qwen2_p128_l15_together.tar +3 -0
  12. qwen2.5_7b_tp/qwen2_p128_l16_together.tar +3 -0
  13. qwen2.5_7b_tp/qwen2_p128_l17_together.tar +3 -0
  14. qwen2.5_7b_tp/qwen2_p128_l18_together.tar +3 -0
  15. qwen2.5_7b_tp/qwen2_p128_l19_together.tar +3 -0
  16. qwen2.5_7b_tp/qwen2_p128_l1_together.tar +3 -0
  17. qwen2.5_7b_tp/qwen2_p128_l20_together.tar +3 -0
  18. qwen2.5_7b_tp/qwen2_p128_l21_together.tar +3 -0
  19. qwen2.5_7b_tp/qwen2_p128_l22_together.tar +3 -0
  20. qwen2.5_7b_tp/qwen2_p128_l23_together.tar +3 -0
  21. qwen2.5_7b_tp/qwen2_p128_l24_together.tar +3 -0
  22. qwen2.5_7b_tp/qwen2_p128_l25_together.tar +3 -0
  23. qwen2.5_7b_tp/qwen2_p128_l26_together.tar +3 -0
  24. qwen2.5_7b_tp/qwen2_p128_l27_together.tar +3 -0
  25. qwen2.5_7b_tp/qwen2_p128_l2_together.tar +3 -0
  26. qwen2.5_7b_tp/qwen2_p128_l3_together.tar +3 -0
  27. qwen2.5_7b_tp/qwen2_p128_l4_together.tar +3 -0
  28. qwen2.5_7b_tp/qwen2_p128_l5_together.tar +3 -0
  29. qwen2.5_7b_tp/qwen2_p128_l6_together.tar +3 -0
  30. qwen2.5_7b_tp/qwen2_p128_l7_together.tar +3 -0
  31. qwen2.5_7b_tp/qwen2_p128_l8_together.tar +3 -0
  32. qwen2.5_7b_tp/qwen2_p128_l9_together.tar +3 -0
  33. qwen2.5_7b_tp/qwen2_post.tar +3 -0
  34. qwen2.5_tokenizer/merges.txt +0 -0
  35. qwen2.5_tokenizer/tokenizer.json +0 -0
  36. qwen2.5_tokenizer/tokenizer_config.json +207 -0
  37. qwen2.5_tokenizer/vocab.json +0 -0
  38. qwen2.5_tokenizer_uid.py +199 -0
  39. run_qwen2.5_7B_axcl_context_tp.sh +12 -0
.gitattributes CHANGED
@@ -33,3 +33,5 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
33
  *.zip filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
 
 
 
33
  *.zip filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
36
+ main_tp_api_x86 filter=lfs diff=lfs merge=lfs -text
37
+ main_tp_x86 filter=lfs diff=lfs merge=lfs -text
main_tp_api_x86 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d1628c4b204088badd37a3a29a36a5d981c124bb0f74280bc2553836d64153d9
3
+ size 1913024
main_tp_x86 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:dc91b50dea818b94e13bc3ef87278c9978a1e78588375775b2c7241945423520
3
+ size 1907088
qwen2.5_7b_tp/model.embed_tokens.weight.bfloat16.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:78ae605b310cf526d54ba22b27796f44010fe28198cb6e3388efb62e09cc2ac4
3
+ size 1089994752
qwen2.5_7b_tp/qwen2_p128_l0_together.tar ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:eb50bee841d4f2c40f1d10deb3cb0ec6986eef37ea87e9977f30e06827383ba4
3
+ size 302888960
qwen2.5_7b_tp/qwen2_p128_l10_together.tar ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:fd7416274ea6254b7829bdeda162cab365941685030feb3c86fec7ec09506d55
3
+ size 302888960
qwen2.5_7b_tp/qwen2_p128_l11_together.tar ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f305680487cbc9b9bd517ca56e1512a5cecbed502d46788734a6829ece31f96e
3
+ size 302888960
qwen2.5_7b_tp/qwen2_p128_l12_together.tar ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5c6c5c2d0ead509055e6642d9ff6663e20a4eb53c4ca2950f0f837ad15ca23cc
3
+ size 302888960
qwen2.5_7b_tp/qwen2_p128_l13_together.tar ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5c9cc5a453b3eba8cf13b7f76fd97ca36585a6eeced75488e02fed1905fe1ce1
3
+ size 302888960
qwen2.5_7b_tp/qwen2_p128_l14_together.tar ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:200907feee14e27d100da7a63b2687383e7966f4416989e09c59d0ff20059642
3
+ size 302888960
qwen2.5_7b_tp/qwen2_p128_l15_together.tar ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:57c264ca8e460c90abf284d9f5883c722e4722144dbcf86d81d1396b8d49ec7d
3
+ size 302888960
qwen2.5_7b_tp/qwen2_p128_l16_together.tar ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7bc99af71e56f868ba0edc878a2f4c5f7192a523866a4562a8595cddcfb3b16c
3
+ size 302888960
qwen2.5_7b_tp/qwen2_p128_l17_together.tar ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7120983258195e25d2587ef5e57d2374d7faf589e8161ced554e7550f00fccfb
3
+ size 302888960
qwen2.5_7b_tp/qwen2_p128_l18_together.tar ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e9d5d36d2cd001ca9e4e8d9fcca5a0bae227d91a935e140a10ff524a77665a65
3
+ size 302888960
qwen2.5_7b_tp/qwen2_p128_l19_together.tar ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9e335a4479e40f28006ef214faa2cccae042819e0bc11d3c07e83fe8967d6f82
3
+ size 302888960
qwen2.5_7b_tp/qwen2_p128_l1_together.tar ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d43e509b0e8fd806cf08853775d46598c0bbaa20b92c102da1cefe57bce3cd09
3
+ size 302888960
qwen2.5_7b_tp/qwen2_p128_l20_together.tar ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:479bfb707ed0824c6f3238604a45695ca758294b2f58d208c5573554ba9b933e
3
+ size 302888960
qwen2.5_7b_tp/qwen2_p128_l21_together.tar ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3e612b41db2d938e15a74c0f7614e228980ba0cee8a46af6f43f088fbd609823
3
+ size 302888960
qwen2.5_7b_tp/qwen2_p128_l22_together.tar ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7d77d5206886bb932395c94da7b222d9ff6413ffadcc8ae31b81672622e1f1b4
3
+ size 302888960
qwen2.5_7b_tp/qwen2_p128_l23_together.tar ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8df5f95091162790f958d912b5388d22edc56463074d0f109629037b9fa9b1ce
3
+ size 302888960
qwen2.5_7b_tp/qwen2_p128_l24_together.tar ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:bfc0d7f800a8a89a4c1181af034f6e6abb8ce1045a807e608db2ada41a86db1e
3
+ size 302888960
qwen2.5_7b_tp/qwen2_p128_l25_together.tar ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:078517c585357c7cdc00629282f4131e3f546024a7f47d14fbe1c8528f0b6814
3
+ size 302888960
qwen2.5_7b_tp/qwen2_p128_l26_together.tar ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:bd2ec02d1011168763b6921a72b34df86336b470ad39d08c5f40f64f7fb2f28f
3
+ size 302888960
qwen2.5_7b_tp/qwen2_p128_l27_together.tar ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9a81c327748f08c12c50be8f3a514739bc23011dc488a949414fefad3e7cf462
3
+ size 302888960
qwen2.5_7b_tp/qwen2_p128_l2_together.tar ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:080fb080648f7cfa405169474011dde3961fb3a595eecd6db464053987334d80
3
+ size 302888960
qwen2.5_7b_tp/qwen2_p128_l3_together.tar ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:22b29fad3e093d465612b307c61f3459e659b6dfc07285b3c95ee8f8ede3b237
3
+ size 302888960
qwen2.5_7b_tp/qwen2_p128_l4_together.tar ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5cf8b8b98472b3b0822a8d7d5b94d206b38fd5d63c2ea35f79c1312ea8e197d8
3
+ size 302888960
qwen2.5_7b_tp/qwen2_p128_l5_together.tar ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7be4e10afb41a504ff1fa07200e0b5f197f70695a2e06c71fed3076c42138315
3
+ size 302888960
qwen2.5_7b_tp/qwen2_p128_l6_together.tar ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:62924b4766657248d79b19c0bf93410e14eaf2e904a9a6569fded7a8a558a775
3
+ size 302888960
qwen2.5_7b_tp/qwen2_p128_l7_together.tar ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7cb84548faaac00205d0fc71907c709a4238aab50db40d1f35c469c21960d013
3
+ size 302888960
qwen2.5_7b_tp/qwen2_p128_l8_together.tar ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f5835511951ab1107c28073a76da850bbe66b4092f7d593c699a362b8bba23de
3
+ size 302888960
qwen2.5_7b_tp/qwen2_p128_l9_together.tar ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:52b59a28e5353f9e6b6ef3a045c27a6926175212b01ab6c54ed2e8b5113b9a02
3
+ size 302888960
qwen2.5_7b_tp/qwen2_post.tar ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:08f605137deadfe376e02efb9412277aaf1b5e096114b92b4422484aef866d57
3
+ size 593192960
qwen2.5_tokenizer/merges.txt ADDED
The diff for this file is too large to render. See raw diff
 
qwen2.5_tokenizer/tokenizer.json ADDED
The diff for this file is too large to render. See raw diff
 
qwen2.5_tokenizer/tokenizer_config.json ADDED
@@ -0,0 +1,207 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "add_prefix_space": false,
3
+ "added_tokens_decoder": {
4
+ "151643": {
5
+ "content": "<|endoftext|>",
6
+ "lstrip": false,
7
+ "normalized": false,
8
+ "rstrip": false,
9
+ "single_word": false,
10
+ "special": true
11
+ },
12
+ "151644": {
13
+ "content": "<|im_start|>",
14
+ "lstrip": false,
15
+ "normalized": false,
16
+ "rstrip": false,
17
+ "single_word": false,
18
+ "special": true
19
+ },
20
+ "151645": {
21
+ "content": "<|im_end|>",
22
+ "lstrip": false,
23
+ "normalized": false,
24
+ "rstrip": false,
25
+ "single_word": false,
26
+ "special": true
27
+ },
28
+ "151646": {
29
+ "content": "<|object_ref_start|>",
30
+ "lstrip": false,
31
+ "normalized": false,
32
+ "rstrip": false,
33
+ "single_word": false,
34
+ "special": true
35
+ },
36
+ "151647": {
37
+ "content": "<|object_ref_end|>",
38
+ "lstrip": false,
39
+ "normalized": false,
40
+ "rstrip": false,
41
+ "single_word": false,
42
+ "special": true
43
+ },
44
+ "151648": {
45
+ "content": "<|box_start|>",
46
+ "lstrip": false,
47
+ "normalized": false,
48
+ "rstrip": false,
49
+ "single_word": false,
50
+ "special": true
51
+ },
52
+ "151649": {
53
+ "content": "<|box_end|>",
54
+ "lstrip": false,
55
+ "normalized": false,
56
+ "rstrip": false,
57
+ "single_word": false,
58
+ "special": true
59
+ },
60
+ "151650": {
61
+ "content": "<|quad_start|>",
62
+ "lstrip": false,
63
+ "normalized": false,
64
+ "rstrip": false,
65
+ "single_word": false,
66
+ "special": true
67
+ },
68
+ "151651": {
69
+ "content": "<|quad_end|>",
70
+ "lstrip": false,
71
+ "normalized": false,
72
+ "rstrip": false,
73
+ "single_word": false,
74
+ "special": true
75
+ },
76
+ "151652": {
77
+ "content": "<|vision_start|>",
78
+ "lstrip": false,
79
+ "normalized": false,
80
+ "rstrip": false,
81
+ "single_word": false,
82
+ "special": true
83
+ },
84
+ "151653": {
85
+ "content": "<|vision_end|>",
86
+ "lstrip": false,
87
+ "normalized": false,
88
+ "rstrip": false,
89
+ "single_word": false,
90
+ "special": true
91
+ },
92
+ "151654": {
93
+ "content": "<|vision_pad|>",
94
+ "lstrip": false,
95
+ "normalized": false,
96
+ "rstrip": false,
97
+ "single_word": false,
98
+ "special": true
99
+ },
100
+ "151655": {
101
+ "content": "<|image_pad|>",
102
+ "lstrip": false,
103
+ "normalized": false,
104
+ "rstrip": false,
105
+ "single_word": false,
106
+ "special": true
107
+ },
108
+ "151656": {
109
+ "content": "<|video_pad|>",
110
+ "lstrip": false,
111
+ "normalized": false,
112
+ "rstrip": false,
113
+ "single_word": false,
114
+ "special": true
115
+ },
116
+ "151657": {
117
+ "content": "<tool_call>",
118
+ "lstrip": false,
119
+ "normalized": false,
120
+ "rstrip": false,
121
+ "single_word": false,
122
+ "special": false
123
+ },
124
+ "151658": {
125
+ "content": "</tool_call>",
126
+ "lstrip": false,
127
+ "normalized": false,
128
+ "rstrip": false,
129
+ "single_word": false,
130
+ "special": false
131
+ },
132
+ "151659": {
133
+ "content": "<|fim_prefix|>",
134
+ "lstrip": false,
135
+ "normalized": false,
136
+ "rstrip": false,
137
+ "single_word": false,
138
+ "special": false
139
+ },
140
+ "151660": {
141
+ "content": "<|fim_middle|>",
142
+ "lstrip": false,
143
+ "normalized": false,
144
+ "rstrip": false,
145
+ "single_word": false,
146
+ "special": false
147
+ },
148
+ "151661": {
149
+ "content": "<|fim_suffix|>",
150
+ "lstrip": false,
151
+ "normalized": false,
152
+ "rstrip": false,
153
+ "single_word": false,
154
+ "special": false
155
+ },
156
+ "151662": {
157
+ "content": "<|fim_pad|>",
158
+ "lstrip": false,
159
+ "normalized": false,
160
+ "rstrip": false,
161
+ "single_word": false,
162
+ "special": false
163
+ },
164
+ "151663": {
165
+ "content": "<|repo_name|>",
166
+ "lstrip": false,
167
+ "normalized": false,
168
+ "rstrip": false,
169
+ "single_word": false,
170
+ "special": false
171
+ },
172
+ "151664": {
173
+ "content": "<|file_sep|>",
174
+ "lstrip": false,
175
+ "normalized": false,
176
+ "rstrip": false,
177
+ "single_word": false,
178
+ "special": false
179
+ }
180
+ },
181
+ "additional_special_tokens": [
182
+ "<|im_start|>",
183
+ "<|im_end|>",
184
+ "<|object_ref_start|>",
185
+ "<|object_ref_end|>",
186
+ "<|box_start|>",
187
+ "<|box_end|>",
188
+ "<|quad_start|>",
189
+ "<|quad_end|>",
190
+ "<|vision_start|>",
191
+ "<|vision_end|>",
192
+ "<|vision_pad|>",
193
+ "<|image_pad|>",
194
+ "<|video_pad|>"
195
+ ],
196
+ "bos_token": null,
197
+ "chat_template": "{%- if tools %}\n {{- '<|im_start|>system\\n' }}\n {%- if messages[0]['role'] == 'system' %}\n {{- messages[0]['content'] }}\n {%- else %}\n {{- 'You are Qwen, created by Alibaba Cloud. You are a helpful assistant.' }}\n {%- endif %}\n {{- \"\\n\\n# 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>\" }}\n {%- for tool in tools %}\n {{- \"\\n\" }}\n {{- tool | tojson }}\n {%- endfor %}\n {{- \"\\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\" }}\n{%- else %}\n {%- if messages[0]['role'] == 'system' %}\n {{- '<|im_start|>system\\n' + messages[0]['content'] + '<|im_end|>\\n' }}\n {%- else %}\n {{- '<|im_start|>system\\nYou are Qwen, created by Alibaba Cloud. You are a helpful assistant.<|im_end|>\\n' }}\n {%- endif %}\n{%- endif %}\n{%- for message in messages %}\n {%- if (message.role == \"user\") or (message.role == \"system\" and not loop.first) or (message.role == \"assistant\" and not message.tool_calls) %}\n {{- '<|im_start|>' + message.role + '\\n' + message.content + '<|im_end|>' + '\\n' }}\n {%- elif message.role == \"assistant\" %}\n {{- '<|im_start|>' + message.role }}\n {%- if message.content %}\n {{- '\\n' + message.content }}\n {%- endif %}\n {%- for tool_call in message.tool_calls %}\n {%- if tool_call.function is defined %}\n {%- set tool_call = tool_call.function %}\n {%- endif %}\n {{- '\\n<tool_call>\\n{\"name\": \"' }}\n {{- tool_call.name }}\n {{- '\", \"arguments\": ' }}\n {{- tool_call.arguments | tojson }}\n {{- '}\\n</tool_call>' }}\n {%- endfor %}\n {{- '<|im_end|>\\n' }}\n {%- elif message.role == \"tool\" %}\n {%- if (loop.index0 == 0) or (messages[loop.index0 - 1].role != \"tool\") %}\n {{- '<|im_start|>user' }}\n {%- endif %}\n {{- '\\n<tool_response>\\n' }}\n {{- message.content }}\n {{- '\\n</tool_response>' }}\n {%- if loop.last or (messages[loop.index0 + 1].role != \"tool\") %}\n {{- '<|im_end|>\\n' }}\n {%- endif %}\n {%- endif %}\n{%- endfor %}\n{%- if add_generation_prompt %}\n {{- '<|im_start|>assistant\\n' }}\n{%- endif %}\n",
198
+ "clean_up_tokenization_spaces": false,
199
+ "eos_token": "<|im_end|>",
200
+ "errors": "replace",
201
+ "model_max_length": 131072,
202
+ "pad_token": "<|endoftext|>",
203
+ "split_special_tokens": false,
204
+ "tokenizer_class": "Qwen2Tokenizer",
205
+ "unk_token": null,
206
+ "add_bos_token": false
207
+ }
qwen2.5_tokenizer/vocab.json ADDED
The diff for this file is too large to render. See raw diff
 
qwen2.5_tokenizer_uid.py ADDED
@@ -0,0 +1,199 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ from transformers import AutoTokenizer, PreTrainedTokenizerFast
2
+ from http.server import HTTPServer, BaseHTTPRequestHandler
3
+ import json
4
+ import argparse
5
+ import uuid
6
+
7
+ # 全局字典:存储 uid 到 Tokenizer_Http 实例的映射
8
+ tokenizers = {}
9
+
10
+ class Tokenizer_Http():
11
+ def __init__(self):
12
+ model_id = "qwen2.5_tokenizer"
13
+ self.tokenizer = AutoTokenizer.from_pretrained(model_id)
14
+ self.messages = [
15
+ {"role": "system", "content": "You are Qwen, created by Alibaba Cloud. You are a helpful assistant."},
16
+ ]
17
+ self.token_ids = []
18
+
19
+ self.token_ids_cache = []
20
+
21
+ def encode(self, prompt, last_reply=None):
22
+ if last_reply is not None:
23
+ self.messages.append({"role": "assistant", "content": last_reply})
24
+ text = self.tokenizer.apply_chat_template(
25
+ self.messages,
26
+ tokenize=False,
27
+ add_generation_prompt=True
28
+ )
29
+ # print("生成的文本:\n============\n", text, "============\n")
30
+ self.token_ids = self.tokenizer.encode(text)[:-3]
31
+ self.messages.append({"role": "user", "content": prompt})
32
+
33
+ text = self.tokenizer.apply_chat_template(
34
+ self.messages,
35
+ tokenize=False,
36
+ add_generation_prompt=True
37
+ )
38
+ print("生成的文本:\n============\n", text, "============\n")
39
+ token_ids = self.tokenizer.encode(text)
40
+ # 找出新增部分
41
+ diff = token_ids[len(self.token_ids):]
42
+ self.token_ids = token_ids
43
+ print(self.decode(diff))
44
+ return token_ids, diff
45
+
46
+ def decode(self, token_ids):
47
+ self.token_ids_cache += token_ids
48
+ text = self.tokenizer.decode(self.token_ids_cache)
49
+ if "\ufffd" in text and len(self.token_ids_cache) < 9:
50
+ print("text 中包含非法字符")
51
+ return ""
52
+ else:
53
+ self.token_ids_cache.clear()
54
+ return text.replace("\ufffd","")
55
+
56
+
57
+ @property
58
+ def bos_id(self):
59
+ return self.tokenizer.bos_token_id
60
+
61
+ @property
62
+ def eos_id(self):
63
+ return self.tokenizer.eos_token_id
64
+
65
+ @property
66
+ def bos_token(self):
67
+ return self.tokenizer.bos_token
68
+
69
+ @property
70
+ def eos_token(self):
71
+ return self.tokenizer.eos_token
72
+
73
+ def reset(self, system_prompt="You are Qwen, created by Alibaba Cloud. You are a helpful assistant."):
74
+ self.messages = [
75
+ {"role": "system", "content": system_prompt},
76
+ ]
77
+ text = self.tokenizer.apply_chat_template(
78
+ self.messages,
79
+ tokenize=False,
80
+ add_generation_prompt=True
81
+ )
82
+ token_ids = self.tokenizer.encode(text)[:-3]
83
+ self.token_ids = token_ids
84
+ print(self.decode(token_ids))
85
+ return token_ids
86
+
87
+
88
+ class Request(BaseHTTPRequestHandler):
89
+ timeout = 5
90
+ server_version = 'Apache'
91
+
92
+ def do_GET(self):
93
+ print("GET 请求路径:", self.path)
94
+ self.send_response(200)
95
+ self.send_header("Content-Type", "application/json")
96
+ self.end_headers()
97
+
98
+ # 新增接口:获取 uid
99
+ if '/get_uid' in self.path:
100
+ new_uid = str(uuid.uuid4())
101
+ print("新 uid:", new_uid)
102
+ # 为该 uid 创建一个新的 Tokenizer_Http 实例
103
+ tokenizers[new_uid] = Tokenizer_Http()
104
+ msg = json.dumps({'uid': new_uid})
105
+ elif '/bos_id' in self.path:
106
+ # 获取 uid 参数(例如 ?uid=xxx)
107
+ uid = self.get_query_param("uid")
108
+ instance: Tokenizer_Http = tokenizers.get(uid)
109
+ if instance is None:
110
+ msg = json.dumps({'error': 'Invalid uid'})
111
+ else:
112
+ bos_id = instance.bos_id
113
+ msg = json.dumps({'bos_id': bos_id if bos_id is not None else -1})
114
+ elif '/eos_id' in self.path:
115
+ uid = self.get_query_param("uid")
116
+ instance: Tokenizer_Http = tokenizers.get(uid)
117
+ if instance is None:
118
+ msg = json.dumps({'error': 'Invalid uid'})
119
+ else:
120
+ eos_id = instance.eos_id
121
+ msg = json.dumps({'eos_id': eos_id if eos_id is not None else -1})
122
+ else:
123
+ msg = json.dumps({'error': 'Invalid GET endpoint'})
124
+
125
+ print("响应消息:", msg)
126
+ self.wfile.write(msg.encode())
127
+
128
+ def do_POST(self):
129
+ content_length = int(self.headers.get('content-length', 0))
130
+ data = self.rfile.read(content_length).decode()
131
+ print("POST 请求路径:", self.path)
132
+ print("接收到的数据:", data)
133
+ req = json.loads(data)
134
+
135
+ self.send_response(200)
136
+ self.send_header("Content-Type", "application/json")
137
+ self.end_headers()
138
+
139
+ if '/encode' in self.path:
140
+ # 请求数据中必须包含 uid, text, 和可选的 last_reply
141
+ uid = req.get('uid')
142
+ prompt = req.get('text')
143
+ last_reply = req.get('last_reply')
144
+ instance: Tokenizer_Http = tokenizers.get(uid)
145
+ if instance is None:
146
+ msg = json.dumps({'error': 'Invalid uid'})
147
+ else:
148
+ token_ids, diff = instance.encode(prompt, last_reply)
149
+ msg = json.dumps({'token_ids': token_ids, 'diff': diff})
150
+ elif '/decode' in self.path:
151
+ uid = req.get('uid')
152
+ token_ids = req.get('token_ids')
153
+ instance: Tokenizer_Http = tokenizers.get(uid)
154
+ if instance is None:
155
+ msg = json.dumps({'error': 'Invalid uid'})
156
+ else:
157
+ text = instance.decode(token_ids)
158
+ msg = json.dumps({'text': text})
159
+ elif '/reset' in self.path:
160
+ uid = req.get("uid")
161
+ system_prompt = req.get("system_prompt")
162
+ instance: Tokenizer_Http = tokenizers.get(uid)
163
+ if instance is None:
164
+ msg = json.dumps({'error': 'Invalid uid'})
165
+ else:
166
+ if system_prompt is not None:
167
+ print("system_prompt:", system_prompt)
168
+ token_ids = instance.reset(system_prompt)
169
+ msg = json.dumps({'token_ids': token_ids})
170
+ else:
171
+ token_ids = instance.reset()
172
+ msg = json.dumps({'token_ids': token_ids})
173
+ else:
174
+ msg = json.dumps({'error': 'Invalid POST endpoint'})
175
+
176
+ print("响应消息:", msg)
177
+ self.wfile.write(msg.encode())
178
+
179
+ def get_query_param(self, key):
180
+ """
181
+ 辅助函数:从 GET 请求的 URL 中获取查询参数的值
182
+ 例如:/bos_id?uid=xxx
183
+ """
184
+ from urllib.parse import urlparse, parse_qs
185
+ query = urlparse(self.path).query
186
+ params = parse_qs(query)
187
+ values = params.get(key)
188
+ return values[0] if values else None
189
+
190
+ if __name__ == "__main__":
191
+ parser = argparse.ArgumentParser()
192
+ parser.add_argument('--host', type=str, default='0.0.0.0')
193
+ parser.add_argument('--port', type=int, default=12345)
194
+ args = parser.parse_args()
195
+
196
+ host = (args.host, args.port)
197
+ print('Server running at http://%s:%s' % host)
198
+ server = HTTPServer(host, Request)
199
+ server.serve_forever()
run_qwen2.5_7B_axcl_context_tp.sh ADDED
@@ -0,0 +1,12 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ./main_tp_x86 \
2
+ --template_filename_axmodel "qwen2.5_7b_tp/qwen2_p128_l%d_together.tar" \
3
+ --axmodel_num 28 \
4
+ --tokenizer_type 2 \
5
+ --url_tokenizer_model "http://127.0.0.1:12345" \
6
+ --filename_post_axmodel "qwen2.5_7b_tp/qwen2_post.tar" \
7
+ --filename_tokens_embed "qwen2.5_7b_tp/model.embed_tokens.weight.bfloat16.bin" \
8
+ --tokens_embed_num 152064 \
9
+ --tokens_embed_size 3584 \
10
+ --use_mmap_load_embed 1 \
11
+ --live_print 1 \
12
+ --devices 0,1,2,3