lbs1163 commited on
Commit
a2e0b8e
·
verified ·
1 Parent(s): e0f6979

chore: set max_batch_size to 1

Browse files

Set max_batch_size according to repository name convention: Batch16 -> 16, Batch32 -> 32, otherwise 1.

Files changed (1) hide show
  1. config.json +41 -40
config.json CHANGED
@@ -1,40 +1,41 @@
1
- {
2
- "architectures": [
3
- "MobilintQwen2ForCausalLM"
4
- ],
5
- "attention_dropout": 0.0,
6
- "auto_map": {
7
- "AutoConfig": "proxy_qwen2.MobilintQwen2Config",
8
- "AutoModelForCausalLM": "proxy_qwen2.MobilintQwen2ForCausalLM"
9
- },
10
- "bos_token_id": 151643,
11
- "eos_token_id": 151645,
12
- "hidden_act": "silu",
13
- "hidden_size": 3584,
14
- "initializer_range": 0.02,
15
- "intermediate_size": 18944,
16
- "max_position_embeddings": 16384,
17
- "max_window_layers": 28,
18
- "model_type": "mobilint-qwen2",
19
- "mxq_path": "Qwen2.5-7B-Instruct-W4V8.mxq",
20
- "num_attention_heads": 28,
21
- "num_hidden_layers": 28,
22
- "num_key_value_heads": 4,
23
- "rms_norm_eps": 1e-06,
24
- "rope_theta": 1000000.0,
25
- "sliding_window": 131072,
26
- "target_cores": [
27
- "0:0"
28
- ],
29
- "tie_word_embeddings": false,
30
- "torch_dtype": "bfloat16",
31
- "transformers_version": "4.43.1",
32
- "use_cache": true,
33
- "use_sliding_window": false,
34
- "vocab_size": 152064,
35
- "npu_prefill_chunk_size": {
36
- "global4": 512,
37
- "global8": 512,
38
- "single": 128
39
- }
40
- }
 
 
1
+ {
2
+ "architectures": [
3
+ "MobilintQwen2ForCausalLM"
4
+ ],
5
+ "attention_dropout": 0.0,
6
+ "auto_map": {
7
+ "AutoConfig": "proxy_qwen2.MobilintQwen2Config",
8
+ "AutoModelForCausalLM": "proxy_qwen2.MobilintQwen2ForCausalLM"
9
+ },
10
+ "bos_token_id": 151643,
11
+ "eos_token_id": 151645,
12
+ "hidden_act": "silu",
13
+ "hidden_size": 3584,
14
+ "initializer_range": 0.02,
15
+ "intermediate_size": 18944,
16
+ "max_position_embeddings": 16384,
17
+ "max_window_layers": 28,
18
+ "model_type": "mobilint-qwen2",
19
+ "mxq_path": "Qwen2.5-7B-Instruct-W4V8.mxq",
20
+ "num_attention_heads": 28,
21
+ "num_hidden_layers": 28,
22
+ "num_key_value_heads": 4,
23
+ "rms_norm_eps": 1e-06,
24
+ "rope_theta": 1000000.0,
25
+ "sliding_window": 131072,
26
+ "target_cores": [
27
+ "0:0"
28
+ ],
29
+ "tie_word_embeddings": false,
30
+ "torch_dtype": "bfloat16",
31
+ "transformers_version": "4.43.1",
32
+ "use_cache": true,
33
+ "use_sliding_window": false,
34
+ "vocab_size": 152064,
35
+ "npu_prefill_chunk_size": {
36
+ "global4": 512,
37
+ "global8": 512,
38
+ "single": 128
39
+ },
40
+ "max_batch_size": 1
41
+ }