lbs1163 commited on
Commit
66c3875
·
verified ·
1 Parent(s): f1ced30

Update config mxq_path for Llama-3.2-1B-Instruct-Batch32-W4V8.mxq

Browse files
Files changed (1) hide show
  1. config.json +61 -61
config.json CHANGED
@@ -1,61 +1,61 @@
1
- {
2
- "architectures": [
3
- "MobilintLlamaForCausalLM"
4
- ],
5
- "attention_bias": false,
6
- "attention_dropout": 0.0,
7
- "auto_map": {
8
- "AutoConfig": "proxy_llama.MobilintLlamaConfig",
9
- "AutoModelForCausalLM": "proxy_llama.MobilintLlamaForCausalLM"
10
- },
11
- "bos_token_id": 128000,
12
- "dev_no": 0,
13
- "eos_token_id": [
14
- 128001,
15
- 128008,
16
- 128009
17
- ],
18
- "head_dim": 64,
19
- "hidden_act": "silu",
20
- "hidden_size": 2048,
21
- "initializer_range": 0.02,
22
- "intermediate_size": 8192,
23
- "max_position_embeddings": 16384,
24
- "mlp_bias": false,
25
- "model_type": "mobilint-llama",
26
- "mxq_path": "Llama-3.2-1B-Instruct-Batch32-W4V8.mxq",
27
- "num_attention_heads": 32,
28
- "num_hidden_layers": 16,
29
- "num_key_value_heads": 8,
30
- "pretraining_tp": 1,
31
- "rms_norm_eps": 1e-05,
32
- "rope_scaling": {
33
- "factor": 32.0,
34
- "high_freq_factor": 4.0,
35
- "low_freq_factor": 1.0,
36
- "original_max_position_embeddings": 2048,
37
- "rope_type": "llama3"
38
- },
39
- "rope_theta": 500000.0,
40
- "target_cores": [
41
- "0:0",
42
- "0:1",
43
- "0:2",
44
- "0:3",
45
- "1:0",
46
- "1:1",
47
- "1:2",
48
- "1:3"
49
- ],
50
- "tie_word_embeddings": false,
51
- "torch_dtype": "bfloat16",
52
- "transformers_version": "4.45.0.dev0",
53
- "use_cache": true,
54
- "vocab_size": 128256,
55
- "npu_prefill_chunk_size": {
56
- "global4": 512,
57
- "global8": 1024,
58
- "single": 256
59
- },
60
- "max_batch_size": 32
61
- }
 
1
+ {
2
+ "architectures": [
3
+ "MobilintLlamaForCausalLM"
4
+ ],
5
+ "attention_bias": false,
6
+ "attention_dropout": 0.0,
7
+ "auto_map": {
8
+ "AutoConfig": "proxy_llama.MobilintLlamaConfig",
9
+ "AutoModelForCausalLM": "proxy_llama.MobilintLlamaForCausalLM"
10
+ },
11
+ "bos_token_id": 128000,
12
+ "dev_no": 0,
13
+ "eos_token_id": [
14
+ 128001,
15
+ 128008,
16
+ 128009
17
+ ],
18
+ "head_dim": 64,
19
+ "hidden_act": "silu",
20
+ "hidden_size": 2048,
21
+ "initializer_range": 0.02,
22
+ "intermediate_size": 8192,
23
+ "max_position_embeddings": 16384,
24
+ "mlp_bias": false,
25
+ "model_type": "mobilint-llama",
26
+ "mxq_path": "Llama-3.2-1B-Instruct-Batch32-W4V8.mxq",
27
+ "num_attention_heads": 32,
28
+ "num_hidden_layers": 16,
29
+ "num_key_value_heads": 8,
30
+ "pretraining_tp": 1,
31
+ "rms_norm_eps": 1e-05,
32
+ "rope_scaling": {
33
+ "factor": 32.0,
34
+ "high_freq_factor": 4.0,
35
+ "low_freq_factor": 1.0,
36
+ "original_max_position_embeddings": 2048,
37
+ "rope_type": "llama3"
38
+ },
39
+ "rope_theta": 500000.0,
40
+ "target_cores": [
41
+ "0:0",
42
+ "0:1",
43
+ "0:2",
44
+ "0:3",
45
+ "1:0",
46
+ "1:1",
47
+ "1:2",
48
+ "1:3"
49
+ ],
50
+ "tie_word_embeddings": false,
51
+ "torch_dtype": "bfloat16",
52
+ "transformers_version": "4.45.0.dev0",
53
+ "use_cache": true,
54
+ "vocab_size": 128256,
55
+ "npu_prefill_chunk_size": {
56
+ "global4": 512,
57
+ "global8": 1024,
58
+ "single": 256
59
+ },
60
+ "max_batch_size": 32
61
+ }