Crystalcareai
/

GemMoE-Base-Hidden

Text Generation

Model card Files Files and versions

Crystalcareai commited on Mar 15, 2024

Commit

3d3abb4

·

verified ·

1 Parent(s): 89644b8

Update config.json

Files changed (1) hide show

config.json +2 -1

config.json CHANGED Viewed

@@ -7,7 +7,8 @@
   "attention_dropout": 0.0,
   "auto_map": {
     "AutoConfig": "Crystalcareai/GemMoE-Beta-1--configuration_gemmoe.GemmoeConfig",
-    "AutoModelForCausalLM": "Crystalcareai/GemMoE-Beta-1--modeling_gemmoe.GemmoeForCausalLM"
   },
   "bos_token_id": 2,
   "eos_token_id": 1,

   "attention_dropout": 0.0,
   "auto_map": {
     "AutoConfig": "Crystalcareai/GemMoE-Beta-1--configuration_gemmoe.GemmoeConfig",
+    "AutoModelForCausalLM": "Crystalcareai/GemMoE-Beta-1--modeling_gemmoe.GemmoeForCausalLM",
+    "AutoTokenizer": "Crystalcareai/GemMoE-Beta-1--tokenization_gemmoe.GemmoeTokenizer"
   },
   "bos_token_id": 2,
   "eos_token_id": 1,