tcheda
/

mot_test

tcheda commited on Jun 27

Commit

02b315d

•

1 Parent(s): 8cd6f66

fix config

Files changed (1) hide show

config.json CHANGED Viewed

@@ -7,7 +7,7 @@
   "auto_map": {
     "AutoConfig": "configuration_mot.MoTConfig",
     "AutoModel": "modeling_mot.MoTModel",
-    "AutoModelForCausalLM": "modeling_mot.MoTForCausalLM"
   },
   "bos_token_id": 50256,
   "embd_pdrop": 0.1,

   "auto_map": {
     "AutoConfig": "configuration_mot.MoTConfig",
     "AutoModel": "modeling_mot.MoTModel",
+    "AutoModelForCausalLM": "modeling_mot.MoTLMHeadModel"
   },
   "bos_token_id": 50256,
   "embd_pdrop": 0.1,