root commited on
Commit
7cf213a
1 Parent(s): e43b1e4

fix config.json to support AutoModelForCausalLM

Browse files
Files changed (1) hide show
  1. config.json +4 -0
config.json CHANGED
@@ -1,5 +1,9 @@
1
  {
2
  "activation_function": "swiglu",
 
 
 
 
3
  "bos_token_id": 13,
4
  "eos_token_id": 13,
5
  "hidden_act": "silu",
 
1
  {
2
  "activation_function": "swiglu",
3
+ "auto_map":{
4
+ "AutoConfig": "configuration_srv1.SRV1Config",
5
+ "AutoModelForCausalLM": "modeling_srv1_tp.SRV1ForCausalLMParallel"
6
+ },
7
  "bos_token_id": 13,
8
  "eos_token_id": 13,
9
  "hidden_act": "silu",