lbs1163 commited on
Commit
422000c
·
verified ·
1 Parent(s): 5af9e5a

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 +46 -45
config.json CHANGED
@@ -1,45 +1,46 @@
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": 100275,
12
- "end_token_id": 100275,
13
- "eos_token_id": 100275,
14
- "head_dim": 128,
15
- "hidden_act": "silu",
16
- "hidden_size": 2048,
17
- "initializer_range": 0.02,
18
- "intermediate_size": 7168,
19
- "max_position_embeddings": 16384,
20
- "mlp_bias": false,
21
- "model_type": "mobilint-llama",
22
- "mxq_path": "HyperCLOVAX-SEED-Text-Instruct-1.5B-W4V8.mxq",
23
- "num_attention_heads": 16,
24
- "num_hidden_layers": 24,
25
- "num_key_value_heads": 8,
26
- "pad_token_id": 100275,
27
- "pretraining_tp": 1,
28
- "resid_pdrop": 0.2,
29
- "rms_norm_eps": 1e-05,
30
- "rope_scaling": null,
31
- "rope_theta": 100000000,
32
- "target_cores": [
33
- "0:0"
34
- ],
35
- "tie_word_embeddings": false,
36
- "torch_dtype": "bfloat16",
37
- "transformers_version": "4.48.2",
38
- "use_cache": true,
39
- "vocab_size": 110592,
40
- "npu_prefill_chunk_size": {
41
- "global4": 512,
42
- "global8": 1024,
43
- "single": 256
44
- }
45
- }
 
 
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": 100275,
12
+ "end_token_id": 100275,
13
+ "eos_token_id": 100275,
14
+ "head_dim": 128,
15
+ "hidden_act": "silu",
16
+ "hidden_size": 2048,
17
+ "initializer_range": 0.02,
18
+ "intermediate_size": 7168,
19
+ "max_position_embeddings": 16384,
20
+ "mlp_bias": false,
21
+ "model_type": "mobilint-llama",
22
+ "mxq_path": "HyperCLOVAX-SEED-Text-Instruct-1.5B-W4V8.mxq",
23
+ "num_attention_heads": 16,
24
+ "num_hidden_layers": 24,
25
+ "num_key_value_heads": 8,
26
+ "pad_token_id": 100275,
27
+ "pretraining_tp": 1,
28
+ "resid_pdrop": 0.2,
29
+ "rms_norm_eps": 1e-05,
30
+ "rope_scaling": null,
31
+ "rope_theta": 100000000,
32
+ "target_cores": [
33
+ "0:0"
34
+ ],
35
+ "tie_word_embeddings": false,
36
+ "torch_dtype": "bfloat16",
37
+ "transformers_version": "4.48.2",
38
+ "use_cache": true,
39
+ "vocab_size": 110592,
40
+ "npu_prefill_chunk_size": {
41
+ "global4": 512,
42
+ "global8": 1024,
43
+ "single": 256
44
+ },
45
+ "max_batch_size": 1
46
+ }