Crystalcareai commited on
Commit
7966564
1 Parent(s): c547bd4

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +5 -6
config.json CHANGED
@@ -1,19 +1,18 @@
1
  {
2
- "_name_or_path": "./mergekit/merged",
3
  "architectures": [
4
  "GemmoeForCausalLM"
5
  ],
6
  "attention_bias": false,
7
  "attention_dropout": 0.0,
8
  "auto_map": {
9
- "AutoConfig": "configuration_gemmoe.GemmoeConfig",
10
- "AutoModelForCausalLM": "modeling_gemmoe.GemmoeForCausalLM"
11
  },
12
  "bos_token_id": 2,
13
  "eos_token_id": 1,
14
  "head_dim": 256,
15
  "hidden_act": "gelu",
16
- "hidden_activation": null,
17
  "hidden_size": 3072,
18
  "initializer_range": 0.02,
19
  "intermediate_size": 24576,
@@ -31,8 +30,8 @@
31
  "rope_theta": 10000.0,
32
  "router_aux_loss_coef": 0.02,
33
  "sliding_window": null,
34
- "torch_dtype": "float16",
35
- "transformers_version": "4.39.0.dev0",
36
  "use_cache": false,
37
  "vocab_size": 256000
38
  }
 
1
  {
2
+ "_name_or_path": "Crystalcareai/GemMoE-Beta-1",
3
  "architectures": [
4
  "GemmoeForCausalLM"
5
  ],
6
  "attention_bias": false,
7
  "attention_dropout": 0.0,
8
  "auto_map": {
9
+ "AutoConfig": "Crystalcareai/GemMoE-Medium-v0.4--configuration_gemmoe.GemmoeConfig",
10
+ "AutoModelForCausalLM": "Crystalcareai/GemMoE-Medium-v0.4--modeling_gemmoe.GemmoeForCausalLM"
11
  },
12
  "bos_token_id": 2,
13
  "eos_token_id": 1,
14
  "head_dim": 256,
15
  "hidden_act": "gelu",
 
16
  "hidden_size": 3072,
17
  "initializer_range": 0.02,
18
  "intermediate_size": 24576,
 
30
  "rope_theta": 10000.0,
31
  "router_aux_loss_coef": 0.02,
32
  "sliding_window": null,
33
+ "torch_dtype": "bfloat16",
34
+ "transformers_version": "4.38.2",
35
  "use_cache": false,
36
  "vocab_size": 256000
37
  }