sino commited on
Commit
896e7f9
1 Parent(s): 73e56d0

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +1 -0
config.json CHANGED
@@ -5,6 +5,7 @@
5
  "auto_map": {
6
  "AutoConfig": "configuration_maelm.MAELMConfig",
7
  "AutoModel": "modeling_maelm.MAEForCausalLM"
 
8
  },
9
  "model_type": "MAELM"
10
  }
 
5
  "auto_map": {
6
  "AutoConfig": "configuration_maelm.MAELMConfig",
7
  "AutoModel": "modeling_maelm.MAEForCausalLM"
8
+ "AutoModelForCausalLM": "modeling_maelm.MAEForCausalLM"
9
  },
10
  "model_type": "MAELM"
11
  }