wli1995 commited on
Commit
eb4deb3
·
verified ·
1 Parent(s): eecd84e

update axmodel and demo

Browse files
Files changed (36) hide show
  1. .gitattributes +1 -1
  2. README.md +1 -1
  3. config.json +51 -0
  4. main_ax650 +3 -0
  5. main_axcl_aarch64 +2 -2
  6. main_axcl_x86 +2 -2
  7. qwen2.5-0.5b-gptq-int4-ax650/model.embed_tokens.weight.bfloat16.bin +1 -1
  8. qwen2.5-0.5b-gptq-int4-ax650/qwen2_p128_l0_together.axmodel +2 -2
  9. qwen2.5-0.5b-gptq-int4-ax650/qwen2_p128_l10_together.axmodel +2 -2
  10. qwen2.5-0.5b-gptq-int4-ax650/qwen2_p128_l11_together.axmodel +2 -2
  11. qwen2.5-0.5b-gptq-int4-ax650/qwen2_p128_l12_together.axmodel +2 -2
  12. qwen2.5-0.5b-gptq-int4-ax650/qwen2_p128_l13_together.axmodel +2 -2
  13. qwen2.5-0.5b-gptq-int4-ax650/qwen2_p128_l14_together.axmodel +2 -2
  14. qwen2.5-0.5b-gptq-int4-ax650/qwen2_p128_l15_together.axmodel +2 -2
  15. qwen2.5-0.5b-gptq-int4-ax650/qwen2_p128_l16_together.axmodel +2 -2
  16. qwen2.5-0.5b-gptq-int4-ax650/qwen2_p128_l17_together.axmodel +2 -2
  17. qwen2.5-0.5b-gptq-int4-ax650/qwen2_p128_l18_together.axmodel +2 -2
  18. qwen2.5-0.5b-gptq-int4-ax650/qwen2_p128_l19_together.axmodel +2 -2
  19. qwen2.5-0.5b-gptq-int4-ax650/qwen2_p128_l1_together.axmodel +2 -2
  20. qwen2.5-0.5b-gptq-int4-ax650/qwen2_p128_l20_together.axmodel +2 -2
  21. qwen2.5-0.5b-gptq-int4-ax650/qwen2_p128_l21_together.axmodel +2 -2
  22. qwen2.5-0.5b-gptq-int4-ax650/qwen2_p128_l22_together.axmodel +2 -2
  23. qwen2.5-0.5b-gptq-int4-ax650/qwen2_p128_l23_together.axmodel +2 -2
  24. qwen2.5-0.5b-gptq-int4-ax650/qwen2_p128_l2_together.axmodel +2 -2
  25. qwen2.5-0.5b-gptq-int4-ax650/qwen2_p128_l3_together.axmodel +2 -2
  26. qwen2.5-0.5b-gptq-int4-ax650/qwen2_p128_l4_together.axmodel +2 -2
  27. qwen2.5-0.5b-gptq-int4-ax650/qwen2_p128_l5_together.axmodel +2 -2
  28. qwen2.5-0.5b-gptq-int4-ax650/qwen2_p128_l6_together.axmodel +2 -2
  29. qwen2.5-0.5b-gptq-int4-ax650/qwen2_p128_l7_together.axmodel +2 -2
  30. qwen2.5-0.5b-gptq-int4-ax650/qwen2_p128_l8_together.axmodel +2 -2
  31. qwen2.5-0.5b-gptq-int4-ax650/qwen2_p128_l9_together.axmodel +2 -2
  32. qwen2.5-0.5b-gptq-int4-ax650/qwen2_post.axmodel +2 -2
  33. qwen2.5_tokenizer_uid.py +199 -0
  34. run_qwen2.5_0.5b_gptq_int4_ax650.sh +5 -8
  35. run_qwen2.5_0.5b_gptq_int4_axcl_aarch64.sh +2 -5
  36. 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: 3.4(Not released yet)
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:bb111fc00c54abb6142a8f44df087bf104c8150a1cefa6be55c6b174b932c4ec
3
- size 999008
 
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:6cba0be8df523e351789fcfa745772423096f52a3b0a760f8d8c9f5b8bb2ec82
3
- size 1022384
 
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:4e96b0df6d274768cbb7e72404011853d23349999b658dc2f4dfb3c431ea223f
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:ef04ff155507e088023f550c844cf18aa3390246015e2a4585b5d7252d884cce
3
- size 9061930
 
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:ee27f1c903adc6dc97223f4a22bcd56f52f688c53412949706a36cf1ff03cb7d
3
- size 9061930
 
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:ad4785e11c1748d2425d1a449034059b3a48910cc258860142d94b97a5de7157
3
- size 9061930
 
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:5da07b1f21d5974f7b80b77204b9981c5ea0d999ddb2874ad797aea9c1643d84
3
- size 9061930
 
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:1750a75cabcdded329b8141a20066741255e0768c8a1e8cd084c0ab10cc5eedf
3
- size 9061930
 
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:a6214f2d1839ef7b8f0d66433e3b5df4e8faef76d31c826a0346959efbbab2e1
3
- size 9061930
 
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:064ad4d2420decc77323bd612e47a249fb884c861facb28e3bba1615c4106d8b
3
- size 9061930
 
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:1c6be483387fa391dd83ab4747106c12e8d4518f5fc125a896d70fc6a68500c3
3
- size 9061930
 
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:f537ef54089b5dc77cde017009cd2c618d0551987a6b1e2f2d9d73686c685a4f
3
- size 9061930
 
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:18af0848c1c8ca5fd9ac797fa449519721cfda08f5f0a5bb0a413a0638d53446
3
- size 9061930
 
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:0563eeda6b066e11a999012e2ae1a8e5c49689afdcc031d20a99c4e94a2804b5
3
- size 9061930
 
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:ddfe42379d84ab9c803bf03da48c316abf907b7ba42dfb17d1a25ce9a6d1e3df
3
- size 9061930
 
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:866d1796031b1b370b2c77d4fe15e33248edad955469d8f05bba9bea59208cac
3
- size 9061930
 
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:5e6e7ddd4533fdd1bcee4be9f492c4572cfafa1bd71f57205178c8567889ccaa
3
- size 9061930
 
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:2fd97d6ea9ccce9fe7c11f11469762a01c4a80c066822d51efd4982fa495e00b
3
- size 9061930
 
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:10390e63263955edd522a544dbee058262979c2995d831e6e75ac2c60f715096
3
- size 9061930
 
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:72bda9ed6a4b256be005f607347345a086a5b528fc724426c81d60175281e8a7
3
- size 9061930
 
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:1f84762b41969a9abcba349b057bd27cce883898eba9db22442cd904a5a16080
3
- size 9061930
 
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:367a6976dac3fbeac8ade0c948608cf97c9362964e78ab3f2381bb0b02a2797a
3
- size 9061930
 
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:3f6b37947e64d3aebf8b70b02ae187a03c8ca776adff199a20f0716f20200bea
3
- size 9061930
 
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:21a801a429a82ecd50fc00bbd68b2df92a505691e7fdbe01fd747f55571d6e01
3
- size 9061930
 
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:f7fedacd5a7ed5ef9d958876885ee38f790891c2b2bdb0937c8a83b7d3fa8e0a
3
- size 9061930
 
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:4c55113e53dc5bcf1dea75d6289dc6d42942cfa6fd2fc836fee16b04cc57e56a
3
- size 9061930
 
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:b4fce55f0e43b31ee933e46899488d194296211f0df55390e89bc2dc232c367c
3
- size 9061930
 
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:0e0533f02e303b9d26ab40a1648917d22d9ae7ac3573a24e4b850fb6d0f71e85
3
- size 147955255
 
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
- ./main_prefill \
 
2
  --template_filename_axmodel "qwen2.5-0.5b-gptq-int4-ax650/qwen2_p128_l%d_together.axmodel" \
3
  --axmodel_num 24 \
4
- --tokenizer_type 2 \
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 1 \
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
- --tokenizer_type 2 \
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
- --continue 1 \
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
- --tokenizer_type 2 \
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
- --continue 1 \
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