KORMo-Team
/

KORMo-10B-sft

Text Generation

Model card Files Files and versions

Update config.json

#1

by BIZLI - opened Nov 4

base: refs/heads/main

←

from: refs/pr/1

Discussion Files changed

Files changed (1) hide show

config.json +1 -1

config.json CHANGED Viewed

@@ -7,7 +7,7 @@
   "auto_map": {
     "AutoConfig": "_configuration_kormo.KORMoConfig",
     "AutoModelForCausalLM": "_modeling_kormo.KORMoForCausalLM",
-    "AutoModel": "_modeling_kormo.KORMoForCausalLM"
   },
   "bos_token_id": 125030,
   "dtype": "bfloat16",

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