Update config.json with auto_map for custom model
Browse files- config.json +6 -2
config.json
CHANGED
|
@@ -38,5 +38,9 @@
|
|
| 38 |
"use_retrieval_augmented": false,
|
| 39 |
"use_rope": true,
|
| 40 |
"use_sliding_window": false,
|
| 41 |
-
"vocab_size": 48000
|
| 42 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 38 |
"use_retrieval_augmented": false,
|
| 39 |
"use_rope": true,
|
| 40 |
"use_sliding_window": false,
|
| 41 |
+
"vocab_size": 48000,
|
| 42 |
+
"auto_map": {
|
| 43 |
+
"AutoConfig": "configuration_openthaiwilai.OpenThaiWilaiConfig",
|
| 44 |
+
"AutoModelForCausalLM": "modeling_openthaiwilai.OpenThaiWilaiForCausalLM"
|
| 45 |
+
}
|
| 46 |
+
}
|