Update config.json
Browse files- config.json +1 -1
config.json
CHANGED
|
@@ -31,7 +31,7 @@
|
|
| 31 |
},
|
| 32 |
"auto_map": {
|
| 33 |
"AutoConfig": "mosaicml/mpt-7b--configuration_mpt.MPTConfig",
|
| 34 |
-
"
|
| 35 |
},
|
| 36 |
"bos_token_id": 0,
|
| 37 |
"d_model": 360,
|
|
|
|
| 31 |
},
|
| 32 |
"auto_map": {
|
| 33 |
"AutoConfig": "mosaicml/mpt-7b--configuration_mpt.MPTConfig",
|
| 34 |
+
"AutoModel": "mosaicml/mpt-7b--modeling_mpt.MPTForCausalLM"
|
| 35 |
},
|
| 36 |
"bos_token_id": 0,
|
| 37 |
"d_model": 360,
|