Crystalcareai
/

Gemma-7b-Fixed

Text Generation

Model card Files Files and versions Community

Crystalcareai commited on Mar 18

Commit

63fd118

•

1 Parent(s): 8a4a739

Update config.json

Files changed (1) hide show

config.json +4 -3

config.json CHANGED Viewed

@@ -1,12 +1,13 @@
 {
   "_name_or_path": "./commerge1",
   "architectures": [
-    "GemmaForCausalLM"
   ],
   "attention_bias": false,
   "attention_dropout": 0.0,
   "auto_map": {
-    "AutoModelForCausalLM": "Crystalcareai/Gemmoe-7b-pre--modeling_gemma.GemmaForCausalLM",
     "AutoTokenizer": "Crystalcareai/GemMoE-Beta-1--tokenization_gemmoe.GemmoeTokenizer"
   },
   "bos_token_id": 2,
@@ -17,7 +18,7 @@
   "initializer_range": 0.02,
   "intermediate_size": 24576,
   "max_position_embeddings": 8192,
-  "model_type": "gemma",
   "num_attention_heads": 16,
   "num_hidden_layers": 28,
   "num_key_value_heads": 16,

 {
   "_name_or_path": "./commerge1",
   "architectures": [
+    "GemmoeForCausalLM"
   ],
   "attention_bias": false,
   "attention_dropout": 0.0,
   "auto_map": {
+    "AutoModelForCausalLM": "Crystalcareai/Gemmoe-7b-pre--modeling_gemmoe.GemmaForCausalLM",
+    "AutoConfig": "Crystalcareai/Gemmoe-7b-pre--configuration_gemmoe.GemmoeConfig",
     "AutoTokenizer": "Crystalcareai/GemMoE-Beta-1--tokenization_gemmoe.GemmoeTokenizer"
   },
   "bos_token_id": 2,
   "initializer_range": 0.02,
   "intermediate_size": 24576,
   "max_position_embeddings": 8192,
+  "model_type": "gemmoe",
   "num_attention_heads": 16,
   "num_hidden_layers": 28,
   "num_key_value_heads": 16,