JonusNattapong commited on
Commit
d808b1c
·
verified ·
1 Parent(s): dbda26d

Update config.json with auto_map for custom model

Browse files
Files changed (1) hide show
  1. 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
+ }