yulan-team
/

YuLan-Mini-Before-Annealing

optimizer_states

Model card Files Files and versions Community

IvanHU commited on 18 days ago

Commit

0bbde8b

·

verified ·

1 Parent(s): 13226fe

Update config.json

Files changed (1) hide show

config.json +2 -2

config.json CHANGED Viewed

@@ -1,11 +1,11 @@
 {
   "architectures": [
-    "YuLanMiniForCausalLM"
   ],
   "auto_map": {
     "AutoConfig": "configuration_yulanmini.YuLanMiniConfig",
     "AutoModel": "modeling_yulanmini.YuLanMiniModel",
-    "AutoModelForCausalLM": "modeling_yulanmini.YuLanMiniForCausalLM",
   },
   "attention_bias": true,
   "attention_dropout": 0.0,

 {
   "architectures": [
+    "YuLanMiniModelForCausalLM"
   ],
   "auto_map": {
     "AutoConfig": "configuration_yulanmini.YuLanMiniConfig",
     "AutoModel": "modeling_yulanmini.YuLanMiniModel",
+    "AutoModelForCausalLM": "modeling_yulanmini.YuLanMiniModelForCausalLM"
   },
   "attention_bias": true,
   "attention_dropout": 0.0,