update axmodel and demo
Browse files- .gitattributes +1 -1
- README.md +1 -1
- config.json +51 -0
- main_ax650 +3 -0
- main_axcl_aarch64 +2 -2
- main_axcl_x86 +2 -2
- qwen2.5-0.5b-gptq-int4-ax650/model.embed_tokens.weight.bfloat16.bin +1 -1
- qwen2.5-0.5b-gptq-int4-ax650/qwen2_p128_l0_together.axmodel +2 -2
- qwen2.5-0.5b-gptq-int4-ax650/qwen2_p128_l10_together.axmodel +2 -2
- qwen2.5-0.5b-gptq-int4-ax650/qwen2_p128_l11_together.axmodel +2 -2
- qwen2.5-0.5b-gptq-int4-ax650/qwen2_p128_l12_together.axmodel +2 -2
- qwen2.5-0.5b-gptq-int4-ax650/qwen2_p128_l13_together.axmodel +2 -2
- qwen2.5-0.5b-gptq-int4-ax650/qwen2_p128_l14_together.axmodel +2 -2
- qwen2.5-0.5b-gptq-int4-ax650/qwen2_p128_l15_together.axmodel +2 -2
- qwen2.5-0.5b-gptq-int4-ax650/qwen2_p128_l16_together.axmodel +2 -2
- qwen2.5-0.5b-gptq-int4-ax650/qwen2_p128_l17_together.axmodel +2 -2
- qwen2.5-0.5b-gptq-int4-ax650/qwen2_p128_l18_together.axmodel +2 -2
- qwen2.5-0.5b-gptq-int4-ax650/qwen2_p128_l19_together.axmodel +2 -2
- qwen2.5-0.5b-gptq-int4-ax650/qwen2_p128_l1_together.axmodel +2 -2
- qwen2.5-0.5b-gptq-int4-ax650/qwen2_p128_l20_together.axmodel +2 -2
- qwen2.5-0.5b-gptq-int4-ax650/qwen2_p128_l21_together.axmodel +2 -2
- qwen2.5-0.5b-gptq-int4-ax650/qwen2_p128_l22_together.axmodel +2 -2
- qwen2.5-0.5b-gptq-int4-ax650/qwen2_p128_l23_together.axmodel +2 -2
- qwen2.5-0.5b-gptq-int4-ax650/qwen2_p128_l2_together.axmodel +2 -2
- qwen2.5-0.5b-gptq-int4-ax650/qwen2_p128_l3_together.axmodel +2 -2
- qwen2.5-0.5b-gptq-int4-ax650/qwen2_p128_l4_together.axmodel +2 -2
- qwen2.5-0.5b-gptq-int4-ax650/qwen2_p128_l5_together.axmodel +2 -2
- qwen2.5-0.5b-gptq-int4-ax650/qwen2_p128_l6_together.axmodel +2 -2
- qwen2.5-0.5b-gptq-int4-ax650/qwen2_p128_l7_together.axmodel +2 -2
- qwen2.5-0.5b-gptq-int4-ax650/qwen2_p128_l8_together.axmodel +2 -2
- qwen2.5-0.5b-gptq-int4-ax650/qwen2_p128_l9_together.axmodel +2 -2
- qwen2.5-0.5b-gptq-int4-ax650/qwen2_post.axmodel +2 -2
- qwen2.5_tokenizer_uid.py +199 -0
- run_qwen2.5_0.5b_gptq_int4_ax650.sh +5 -8
- run_qwen2.5_0.5b_gptq_int4_axcl_aarch64.sh +2 -5
- run_qwen2.5_0.5b_gptq_int4_axcl_x86.sh +2 -5
.gitattributes
CHANGED
@@ -37,4 +37,4 @@ main_axcl_aarch64 filter=lfs diff=lfs merge=lfs -text
|
|
37 |
main_axcl_x86 filter=lfs diff=lfs merge=lfs -text
|
38 |
main_prefill filter=lfs diff=lfs merge=lfs -text
|
39 |
qwen2.5-0.5b-gptq-int4-ax650/qwen2_post.axmodel filter=lfs diff=lfs merge=lfs -text
|
40 |
-
*.axmodel filter=lfs diff=lfs merge=lfs -text
|
|
|
37 |
main_axcl_x86 filter=lfs diff=lfs merge=lfs -text
|
38 |
main_prefill filter=lfs diff=lfs merge=lfs -text
|
39 |
qwen2.5-0.5b-gptq-int4-ax650/qwen2_post.axmodel filter=lfs diff=lfs merge=lfs -text
|
40 |
+
*.axmodel filter=lfs diff=lfs merge=lfs -textmain_ax650 filter=lfs diff=lfs merge=lfs -text
|
README.md
CHANGED
@@ -17,7 +17,7 @@ This version of Qwen2.5-0.5B-Instruct-GPTQ-Int4 has been converted to run on the
|
|
17 |
|
18 |
This model has been optimized with the following LoRA:
|
19 |
|
20 |
-
Compatible with Pulsar2 version:
|
21 |
|
22 |
## Convert tools links:
|
23 |
|
|
|
17 |
|
18 |
This model has been optimized with the following LoRA:
|
19 |
|
20 |
+
Compatible with Pulsar2 version: 4.2(Not released yet)
|
21 |
|
22 |
## Convert tools links:
|
23 |
|
config.json
CHANGED
@@ -0,0 +1,51 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"architectures": [
|
3 |
+
"Qwen2ForCausalLM"
|
4 |
+
],
|
5 |
+
"attention_dropout": 0.0,
|
6 |
+
"bos_token_id": 151643,
|
7 |
+
"eos_token_id": 151645,
|
8 |
+
"hidden_act": "silu",
|
9 |
+
"hidden_size": 896,
|
10 |
+
"initializer_range": 0.02,
|
11 |
+
"intermediate_size": 4864,
|
12 |
+
"max_position_embeddings": 32768,
|
13 |
+
"max_window_layers": 21,
|
14 |
+
"model_type": "qwen2",
|
15 |
+
"num_attention_heads": 14,
|
16 |
+
"num_hidden_layers": 24,
|
17 |
+
"num_key_value_heads": 2,
|
18 |
+
"quantization_config": {
|
19 |
+
"batch_size": 1,
|
20 |
+
"bits": 4,
|
21 |
+
"block_name_to_quantize": null,
|
22 |
+
"cache_block_outputs": true,
|
23 |
+
"damp_percent": 0.01,
|
24 |
+
"dataset": null,
|
25 |
+
"desc_act": false,
|
26 |
+
"exllama_config": {
|
27 |
+
"version": 1
|
28 |
+
},
|
29 |
+
"group_size": 128,
|
30 |
+
"max_input_length": null,
|
31 |
+
"model_seqlen": null,
|
32 |
+
"module_name_preceding_first_block": null,
|
33 |
+
"modules_in_block_to_quantize": null,
|
34 |
+
"pad_token_id": null,
|
35 |
+
"quant_method": "gptq",
|
36 |
+
"sym": true,
|
37 |
+
"tokenizer": null,
|
38 |
+
"true_sequential": true,
|
39 |
+
"use_cuda_fp16": false,
|
40 |
+
"use_exllama": true
|
41 |
+
},
|
42 |
+
"rms_norm_eps": 1e-06,
|
43 |
+
"rope_theta": 1000000.0,
|
44 |
+
"sliding_window": 32768,
|
45 |
+
"tie_word_embeddings": true,
|
46 |
+
"torch_dtype": "float16",
|
47 |
+
"transformers_version": "4.39.3",
|
48 |
+
"use_cache": true,
|
49 |
+
"use_sliding_window": false,
|
50 |
+
"vocab_size": 151936
|
51 |
+
}
|
main_ax650
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:f19ddeb193769b16aa8c5d9bba887558aa0a4ed10eb50a19d9bc117f1ba527e5
|
3 |
+
size 985352
|
main_axcl_aarch64
CHANGED
@@ -1,3 +1,3 @@
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
-
oid sha256:
|
3 |
-
size
|
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:1f9f1a1ca329b47f70840e8b6d104ce8248a82326aa2402bccb31144590a8fb2
|
3 |
+
size 1725008
|
main_axcl_x86
CHANGED
@@ -1,3 +1,3 @@
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
-
oid sha256:
|
3 |
-
size
|
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:928d36be31c15d081a7d346464f41458e9624d8b68d5f7dfb3d3189686ce2754
|
3 |
+
size 8421624
|
qwen2.5-0.5b-gptq-int4-ax650/model.embed_tokens.weight.bfloat16.bin
CHANGED
@@ -1,3 +1,3 @@
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
-
oid sha256:
|
3 |
size 272269312
|
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:4a00e7baebfd01831d3104348802fb85cc075365a86b6b0007515fa1b9225e7d
|
3 |
size 272269312
|
qwen2.5-0.5b-gptq-int4-ax650/qwen2_p128_l0_together.axmodel
CHANGED
@@ -1,3 +1,3 @@
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
-
oid sha256:
|
3 |
-
size
|
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:c335becab00a3e60c0c11c1fae0d97e5945b92bf5555b047fbf3d3d45f5fa06e
|
3 |
+
size 12286102
|
qwen2.5-0.5b-gptq-int4-ax650/qwen2_p128_l10_together.axmodel
CHANGED
@@ -1,3 +1,3 @@
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
-
oid sha256:
|
3 |
-
size
|
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:dc334ccd85edc693eae0793d38aaca9bfeb411099b88f24af87396b9e2d3dd7f
|
3 |
+
size 12286102
|
qwen2.5-0.5b-gptq-int4-ax650/qwen2_p128_l11_together.axmodel
CHANGED
@@ -1,3 +1,3 @@
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
-
oid sha256:
|
3 |
-
size
|
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:d7e36187af29691c533346b0e20ef7575f89ffa48783b323827f58e4c0b4963a
|
3 |
+
size 12286102
|
qwen2.5-0.5b-gptq-int4-ax650/qwen2_p128_l12_together.axmodel
CHANGED
@@ -1,3 +1,3 @@
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
-
oid sha256:
|
3 |
-
size
|
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:e7a755f6a4121ee94f1903620237ae195a8628702fdb50de26162f536c399d07
|
3 |
+
size 12286102
|
qwen2.5-0.5b-gptq-int4-ax650/qwen2_p128_l13_together.axmodel
CHANGED
@@ -1,3 +1,3 @@
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
-
oid sha256:
|
3 |
-
size
|
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:a8216164c2518651e754e26dac698e4a85d677742d01afe367844931f877083c
|
3 |
+
size 12286102
|
qwen2.5-0.5b-gptq-int4-ax650/qwen2_p128_l14_together.axmodel
CHANGED
@@ -1,3 +1,3 @@
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
-
oid sha256:
|
3 |
-
size
|
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:ae8e1e935534654142c588b9ee5fef648eb83559d83f1391d4af148e9fcf0505
|
3 |
+
size 12286102
|
qwen2.5-0.5b-gptq-int4-ax650/qwen2_p128_l15_together.axmodel
CHANGED
@@ -1,3 +1,3 @@
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
-
oid sha256:
|
3 |
-
size
|
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:9975bb26552f7983515417fe8d2338d33a4d2d960d393e9e6833192992fc8bea
|
3 |
+
size 12286102
|
qwen2.5-0.5b-gptq-int4-ax650/qwen2_p128_l16_together.axmodel
CHANGED
@@ -1,3 +1,3 @@
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
-
oid sha256:
|
3 |
-
size
|
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:7713922f7c7c80cadb9aef4e5d8edac42e5f667ad9e33fdf37390ce97da3c5f5
|
3 |
+
size 12286102
|
qwen2.5-0.5b-gptq-int4-ax650/qwen2_p128_l17_together.axmodel
CHANGED
@@ -1,3 +1,3 @@
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
-
oid sha256:
|
3 |
-
size
|
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:bec5406b9c1f2b4f7909e1b6614e29a9653ae754f29518b7abf072a24620b6fc
|
3 |
+
size 12286102
|
qwen2.5-0.5b-gptq-int4-ax650/qwen2_p128_l18_together.axmodel
CHANGED
@@ -1,3 +1,3 @@
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
-
oid sha256:
|
3 |
-
size
|
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:6ae94ef72d6b4b7a60eb7c40aab314e8d587f3e88455e39a2b700e7dd2c03205
|
3 |
+
size 12286102
|
qwen2.5-0.5b-gptq-int4-ax650/qwen2_p128_l19_together.axmodel
CHANGED
@@ -1,3 +1,3 @@
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
-
oid sha256:
|
3 |
-
size
|
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:9bb02c80c9ddc8bd251e064b44ca17a8ce516e845038aca79ed6c9e24727114f
|
3 |
+
size 12286102
|
qwen2.5-0.5b-gptq-int4-ax650/qwen2_p128_l1_together.axmodel
CHANGED
@@ -1,3 +1,3 @@
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
-
oid sha256:
|
3 |
-
size
|
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:f6a080891672734fc64036bd629477719e2569a6afda3e4e316b61ff86364c83
|
3 |
+
size 12286102
|
qwen2.5-0.5b-gptq-int4-ax650/qwen2_p128_l20_together.axmodel
CHANGED
@@ -1,3 +1,3 @@
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
-
oid sha256:
|
3 |
-
size
|
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:69f3be8f72b6eef91ac0cf91744afe085e148e4fa1f6c4d7b0d718ffb9f469fe
|
3 |
+
size 12286102
|
qwen2.5-0.5b-gptq-int4-ax650/qwen2_p128_l21_together.axmodel
CHANGED
@@ -1,3 +1,3 @@
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
-
oid sha256:
|
3 |
-
size
|
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:6c6cf9c7079bc1758e9f7591b95899c8bbeec6fb2e33f5f512913f4f9a8441ad
|
3 |
+
size 12286102
|
qwen2.5-0.5b-gptq-int4-ax650/qwen2_p128_l22_together.axmodel
CHANGED
@@ -1,3 +1,3 @@
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
-
oid sha256:
|
3 |
-
size
|
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:8019c3f30ef46f4a04c59e35b3522f1f6692c448de8785325280c6cc5ebdccea
|
3 |
+
size 12286102
|
qwen2.5-0.5b-gptq-int4-ax650/qwen2_p128_l23_together.axmodel
CHANGED
@@ -1,3 +1,3 @@
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
-
oid sha256:
|
3 |
-
size
|
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:a2985914bbb2b87fa0fb763ed7a1216af94ec3e8db141ee4e58cd30e802d9a28
|
3 |
+
size 12286102
|
qwen2.5-0.5b-gptq-int4-ax650/qwen2_p128_l2_together.axmodel
CHANGED
@@ -1,3 +1,3 @@
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
-
oid sha256:
|
3 |
-
size
|
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:37f1e0a303566f121346af3f88209f2bdffff62b77d1276a8c6b607f92661a2c
|
3 |
+
size 12286102
|
qwen2.5-0.5b-gptq-int4-ax650/qwen2_p128_l3_together.axmodel
CHANGED
@@ -1,3 +1,3 @@
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
-
oid sha256:
|
3 |
-
size
|
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:1396fe073b577feb67af43b0d40a7dfbeeea646d29d73d5ea2250137b4041c61
|
3 |
+
size 12286102
|
qwen2.5-0.5b-gptq-int4-ax650/qwen2_p128_l4_together.axmodel
CHANGED
@@ -1,3 +1,3 @@
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
-
oid sha256:
|
3 |
-
size
|
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:e27b2ad022d2c688aba65ba1582cdfc1225f3021af53d6f9eec433d6128733c5
|
3 |
+
size 12286102
|
qwen2.5-0.5b-gptq-int4-ax650/qwen2_p128_l5_together.axmodel
CHANGED
@@ -1,3 +1,3 @@
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
-
oid sha256:
|
3 |
-
size
|
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:45c93a96f4ebcff7a7b01d229946645346ba40b1d340f4d66d97bf8dee7c5827
|
3 |
+
size 12286102
|
qwen2.5-0.5b-gptq-int4-ax650/qwen2_p128_l6_together.axmodel
CHANGED
@@ -1,3 +1,3 @@
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
-
oid sha256:
|
3 |
-
size
|
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:7d580d6a8cdfdc270a1a76276a1c4922a638b01aef7d841c7d291f425873cc35
|
3 |
+
size 12286102
|
qwen2.5-0.5b-gptq-int4-ax650/qwen2_p128_l7_together.axmodel
CHANGED
@@ -1,3 +1,3 @@
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
-
oid sha256:
|
3 |
-
size
|
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:80620239eee1edb2980900d8b97c6ed9b2c27e9e0b8e97368df3ab17b8e785a9
|
3 |
+
size 12286102
|
qwen2.5-0.5b-gptq-int4-ax650/qwen2_p128_l8_together.axmodel
CHANGED
@@ -1,3 +1,3 @@
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
-
oid sha256:
|
3 |
-
size
|
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:6e910d13e4e88dd3ad0c7461af7d6151e877bb08fd730957225b2eb0a9743172
|
3 |
+
size 12286102
|
qwen2.5-0.5b-gptq-int4-ax650/qwen2_p128_l9_together.axmodel
CHANGED
@@ -1,3 +1,3 @@
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
-
oid sha256:
|
3 |
-
size
|
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:00b8126bed769a8f7cfc7e5b4bf4239e111e03168aa15e10bf83c959e4e31cfa
|
3 |
+
size 12286102
|
qwen2.5-0.5b-gptq-int4-ax650/qwen2_post.axmodel
CHANGED
@@ -1,3 +1,3 @@
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
-
oid sha256:
|
3 |
-
size
|
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:af27ff530b80e60e1472c89017bb9c981f76960f6b01ac18aa32b3c83f7511fb
|
3 |
+
size 147954839
|
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:
|
50 |
+
print("text 中包含非法字符")
|
51 |
+
return ""
|
52 |
+
else:
|
53 |
+
self.token_ids_cache.clear()
|
54 |
+
return text
|
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_0.5b_gptq_int4_ax650.sh
CHANGED
@@ -1,14 +1,11 @@
|
|
1 |
-
./
|
|
|
2 |
--template_filename_axmodel "qwen2.5-0.5b-gptq-int4-ax650/qwen2_p128_l%d_together.axmodel" \
|
3 |
--axmodel_num 24 \
|
4 |
-
--
|
5 |
-
--filename_tokenizer_model "http://127.0.0.1:12345" \
|
6 |
-
--bos 0 --eos 0 \
|
7 |
--filename_post_axmodel "qwen2.5-0.5b-gptq-int4-ax650/qwen2_post.axmodel" \
|
8 |
--filename_tokens_embed "qwen2.5-0.5b-gptq-int4-ax650/model.embed_tokens.weight.bfloat16.bin" \
|
9 |
--tokens_embed_num 151936 \
|
10 |
--tokens_embed_size 896 \
|
11 |
-
--use_mmap_load_embed
|
12 |
-
--live_print 1
|
13 |
-
--continue 1 \
|
14 |
-
--prompt "$1"
|
|
|
1 |
+
./main_ax650 \
|
2 |
+
--system_prompt "You are Qwen, created by Alibaba Cloud. You are a helpful assistant." \
|
3 |
--template_filename_axmodel "qwen2.5-0.5b-gptq-int4-ax650/qwen2_p128_l%d_together.axmodel" \
|
4 |
--axmodel_num 24 \
|
5 |
+
--url_tokenizer_model "http://127.0.0.1:12345" \
|
|
|
|
|
6 |
--filename_post_axmodel "qwen2.5-0.5b-gptq-int4-ax650/qwen2_post.axmodel" \
|
7 |
--filename_tokens_embed "qwen2.5-0.5b-gptq-int4-ax650/model.embed_tokens.weight.bfloat16.bin" \
|
8 |
--tokens_embed_num 151936 \
|
9 |
--tokens_embed_size 896 \
|
10 |
+
--use_mmap_load_embed 0 \
|
11 |
+
--live_print 1
|
|
|
|
run_qwen2.5_0.5b_gptq_int4_axcl_aarch64.sh
CHANGED
@@ -1,14 +1,11 @@
|
|
1 |
./main_axcl_aarch64 \
|
2 |
--template_filename_axmodel "qwen2.5-0.5b-gptq-int4-ax650/qwen2_p128_l%d_together.axmodel" \
|
3 |
--axmodel_num 24 \
|
4 |
-
--
|
5 |
-
--filename_tokenizer_model "http://127.0.0.1:12345" \
|
6 |
-
--bos 0 --eos 0 \
|
7 |
--filename_post_axmodel "qwen2.5-0.5b-gptq-int4-ax650/qwen2_post.axmodel" \
|
8 |
--filename_tokens_embed "qwen2.5-0.5b-gptq-int4-ax650/model.embed_tokens.weight.bfloat16.bin" \
|
9 |
--tokens_embed_num 151936 \
|
10 |
--tokens_embed_size 896 \
|
11 |
--use_mmap_load_embed 0 \
|
12 |
--live_print 1 \
|
13 |
-
--
|
14 |
-
--prompt "$1"
|
|
|
1 |
./main_axcl_aarch64 \
|
2 |
--template_filename_axmodel "qwen2.5-0.5b-gptq-int4-ax650/qwen2_p128_l%d_together.axmodel" \
|
3 |
--axmodel_num 24 \
|
4 |
+
--url_tokenizer_model "http://127.0.0.1:12345" \
|
|
|
|
|
5 |
--filename_post_axmodel "qwen2.5-0.5b-gptq-int4-ax650/qwen2_post.axmodel" \
|
6 |
--filename_tokens_embed "qwen2.5-0.5b-gptq-int4-ax650/model.embed_tokens.weight.bfloat16.bin" \
|
7 |
--tokens_embed_num 151936 \
|
8 |
--tokens_embed_size 896 \
|
9 |
--use_mmap_load_embed 0 \
|
10 |
--live_print 1 \
|
11 |
+
--devices 0
|
|
run_qwen2.5_0.5b_gptq_int4_axcl_x86.sh
CHANGED
@@ -1,14 +1,11 @@
|
|
1 |
./main_axcl_x86 \
|
2 |
--template_filename_axmodel "qwen2.5-0.5b-gptq-int4-ax650/qwen2_p128_l%d_together.axmodel" \
|
3 |
--axmodel_num 24 \
|
4 |
-
--
|
5 |
-
--filename_tokenizer_model "http://127.0.0.1:12345" \
|
6 |
-
--bos 0 --eos 0 \
|
7 |
--filename_post_axmodel "qwen2.5-0.5b-gptq-int4-ax650/qwen2_post.axmodel" \
|
8 |
--filename_tokens_embed "qwen2.5-0.5b-gptq-int4-ax650/model.embed_tokens.weight.bfloat16.bin" \
|
9 |
--tokens_embed_num 151936 \
|
10 |
--tokens_embed_size 896 \
|
11 |
--use_mmap_load_embed 0 \
|
12 |
--live_print 1 \
|
13 |
-
--
|
14 |
-
--prompt "$1"
|
|
|
1 |
./main_axcl_x86 \
|
2 |
--template_filename_axmodel "qwen2.5-0.5b-gptq-int4-ax650/qwen2_p128_l%d_together.axmodel" \
|
3 |
--axmodel_num 24 \
|
4 |
+
--url_tokenizer_model "http://127.0.0.1:12345" \
|
|
|
|
|
5 |
--filename_post_axmodel "qwen2.5-0.5b-gptq-int4-ax650/qwen2_post.axmodel" \
|
6 |
--filename_tokens_embed "qwen2.5-0.5b-gptq-int4-ax650/model.embed_tokens.weight.bfloat16.bin" \
|
7 |
--tokens_embed_num 151936 \
|
8 |
--tokens_embed_size 896 \
|
9 |
--use_mmap_load_embed 0 \
|
10 |
--live_print 1 \
|
11 |
+
--devices 0
|
|