michaelbzhu commited on
Commit
4fccf20
·
verified ·
1 Parent(s): 328b71b

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +1 -1
config.json CHANGED
@@ -1,6 +1,6 @@
1
  {
2
  "architectures": [
3
- "CustomModel"
4
  ],
5
  "auto_map": {
6
  "AutoConfig": "configuration_custom_mbz_test.CustomConfig",
 
1
  {
2
  "architectures": [
3
+ "CustomModelForCausalLM"
4
  ],
5
  "auto_map": {
6
  "AutoConfig": "configuration_custom_mbz_test.CustomConfig",