SeaWolf-AI commited on
Commit
eea4d0f
·
verified ·
1 Parent(s): 613f399

fix: make model loadable via AutoModelForCausalLM (add auto_map, flatten module files to repo root, inherit GenerationMixin)

Browse files
Files changed (1) hide show
  1. config.json +51 -46
config.json CHANGED
@@ -1,47 +1,52 @@
1
- {
2
- "vocab_size": 151936,
3
- "hidden_size": 2048,
4
- "intermediate_size": 6144,
5
- "num_hidden_layers": 49,
6
- "num_attention_heads": 16,
7
- "num_key_value_heads": 4,
8
- "head_dim": 128,
9
- "attention_dropout": 0.0,
10
- "sliding_window_size": 512,
11
- "compress_block_size": 16,
12
- "num_experts": 25,
13
- "num_experts_per_tok": 7,
14
- "expert_intermediate_size": 640,
15
- "use_shared_expert": true,
16
- "router_aux_loss_coef": 0.001,
17
- "max_position_embeddings": 4096,
18
- "rope_theta": 10000.0,
19
- "rms_norm_eps": 1e-06,
20
- "hidden_act": "silu",
21
- "initializer_range": 0.02,
22
- "use_cache": false,
23
- "output_router_logits": true,
24
- "return_dict": true,
25
- "output_hidden_states": false,
26
- "dtype": "bfloat16",
27
- "chunk_size_feed_forward": 0,
28
- "is_encoder_decoder": false,
29
- "architectures": [
30
- "AETHERV27wayForCausalLM"
31
- ],
32
- "id2label": {
33
- "0": "LABEL_0",
34
- "1": "LABEL_1"
35
- },
36
- "label2id": {
37
- "LABEL_0": 0,
38
- "LABEL_1": 1
39
- },
40
- "problem_type": null,
41
- "_name_or_path": "",
42
- "transformers_version": "5.3.0",
43
- "pad_token_id": 151643,
44
- "tie_word_embeddings": false,
45
- "model_type": "aether_v2_7way",
46
- "output_attentions": false
 
 
 
 
 
47
  }
 
1
+ {
2
+ "vocab_size": 151936,
3
+ "hidden_size": 2048,
4
+ "intermediate_size": 6144,
5
+ "num_hidden_layers": 49,
6
+ "num_attention_heads": 16,
7
+ "num_key_value_heads": 4,
8
+ "head_dim": 128,
9
+ "attention_dropout": 0.0,
10
+ "sliding_window_size": 512,
11
+ "compress_block_size": 16,
12
+ "num_experts": 25,
13
+ "num_experts_per_tok": 7,
14
+ "expert_intermediate_size": 640,
15
+ "use_shared_expert": true,
16
+ "router_aux_loss_coef": 0.001,
17
+ "max_position_embeddings": 4096,
18
+ "rope_theta": 10000.0,
19
+ "rms_norm_eps": 1e-06,
20
+ "hidden_act": "silu",
21
+ "initializer_range": 0.02,
22
+ "use_cache": false,
23
+ "output_router_logits": true,
24
+ "return_dict": true,
25
+ "output_hidden_states": false,
26
+ "dtype": "bfloat16",
27
+ "chunk_size_feed_forward": 0,
28
+ "is_encoder_decoder": false,
29
+ "architectures": [
30
+ "AETHERV27wayForCausalLM"
31
+ ],
32
+ "id2label": {
33
+ "0": "LABEL_0",
34
+ "1": "LABEL_1"
35
+ },
36
+ "label2id": {
37
+ "LABEL_0": 0,
38
+ "LABEL_1": 1
39
+ },
40
+ "problem_type": null,
41
+ "_name_or_path": "",
42
+ "transformers_version": "5.3.0",
43
+ "pad_token_id": 151643,
44
+ "tie_word_embeddings": false,
45
+ "model_type": "aether_v2_7way",
46
+ "output_attentions": false,
47
+ "auto_map": {
48
+ "AutoConfig": "configuration_aether_v2_7way.AETHERV27wayConfig",
49
+ "AutoModelForCausalLM": "modeling_aether_v2_7way.AETHERV27wayForCausalLM",
50
+ "AutoModel": "modeling_aether_v2_7way.AETHERV27wayModel"
51
+ }
52
  }