BoyongWu commited on
Commit
ae7ebb9
·
verified ·
1 Parent(s): 099dcc7

Update config.json

Browse files

Add "AutoModelForCausalLM" for SFT

Files changed (1) hide show
  1. config.json +2 -1
config.json CHANGED
@@ -3,7 +3,8 @@
3
  "Step1ForCausalLM"
4
  ],
5
  "auto_map": {
6
- "AutoConfig": "configuration_step1.Step1Config"
 
7
  },
8
  "model_type": "step1",
9
  "bos_token_id": 1,
 
3
  "Step1ForCausalLM"
4
  ],
5
  "auto_map": {
6
+ "AutoConfig": "configuration_step1.Step1Config",
7
+ "AutoModelForCausalLM": "modeling_step1.Step1ForCausalLM"
8
  },
9
  "model_type": "step1",
10
  "bos_token_id": 1,