KORMo-Team
/

KORMo-10B-base

Text Generation

Model card Files Files and versions

Update config.json

#1

by BIZLI - opened 25 days ago

base: refs/heads/main

←

from: refs/pr/1

Discussion Files changed

Files changed (1) hide show

config.json +1 -0

config.json CHANGED Viewed

@@ -6,6 +6,7 @@
   "attention_dropout": 0.0,
   "auto_map": {
     "AutoConfig": "_configuration_kormo.KORMoConfig",
     "AutoModelForCausalLM": "_modeling_kormo.KORMoForCausalLM"
   },
   "bos_token_id": 125030,

   "attention_dropout": 0.0,
   "auto_map": {
     "AutoConfig": "_configuration_kormo.KORMoConfig",
+    "AutoModel": "_modeling_kormo.KORMoModel",
     "AutoModelForCausalLM": "_modeling_kormo.KORMoForCausalLM"
   },
   "bos_token_id": 125030,