reedhs commited on
Commit
4ceb61e
·
1 Parent(s): 20a92d2
Files changed (1) hide show
  1. config.json +3 -3
config.json CHANGED
@@ -1,11 +1,11 @@
1
  {
2
- "_name_or_path": "baichuan-inc/baichuan-7B",
3
  "architectures": [
4
  "BaiChuanForCausalLM"
5
  ],
6
  "auto_map": {
7
- "AutoConfig": "baichuan-inc/baichuan-7B--configuration_baichuan.BaiChuanConfig",
8
- "AutoModelForCausalLM": "baichuan-inc/baichuan-7B--modeling_baichuan.BaiChuanForCausalLM"
9
  },
10
  "bos_token_id": 1,
11
  "eos_token_id": 2,
 
1
  {
2
+ "_name_or_path": "reedhs/baichuan-7b-sharded",
3
  "architectures": [
4
  "BaiChuanForCausalLM"
5
  ],
6
  "auto_map": {
7
+ "AutoConfig": "configuration_baichuan.BaiChuanConfig",
8
+ "AutoModelForCausalLM": "modeling_baichuan.BaiChuanForCausalLM"
9
  },
10
  "bos_token_id": 1,
11
  "eos_token_id": 2,