abeat commited on
Commit
2345883
·
verified ·
1 Parent(s): 0db9034

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +2 -2
config.json CHANGED
@@ -4,8 +4,8 @@
4
  "MyCustomModelForCausalLM"
5
  ],
6
  "auto_map": {
7
- "AutoConfig": "blahblahtron_1_3B.HFWrapperConfig",
8
- "AutoModelForCausalLM": "blahblahtron_1_3B.MyCustomModelForCausalLM"
9
  },
10
  "bos_token_id": 50256,
11
  "dropout": 0.1,
 
4
  "MyCustomModelForCausalLM"
5
  ],
6
  "auto_map": {
7
+ "AutoConfig": "blahblahtron_1_1B.HFWrapperConfig",
8
+ "AutoModelForCausalLM": "blahblahtron_1_1B.MyCustomModelForCausalLM"
9
  },
10
  "bos_token_id": 50256,
11
  "dropout": 0.1,