aframson commited on
Commit
d2e4408
1 Parent(s): f514d5b
Files changed (1) hide show
  1. config.json +5 -5
config.json CHANGED
@@ -1,12 +1,12 @@
1
  {
2
  "architectures": [
3
- "GPT"
4
  ],
5
  "auto_map": {
6
- "AutoConfig": "model.GPTConfig",
7
- "AutoModel": "model.GPT",
8
- "AutoModelForCausalLM": "model.GPT",
9
- "AutoModelForQuestionAnswering": "model.GPT"
10
  },
11
  "batch_size": 12,
12
  "block_size": 100,
 
1
  {
2
  "architectures": [
3
+ "GPT2LMHeadModel"
4
  ],
5
  "auto_map": {
6
+ "AutoConfig": "GPT2Tokenizer",
7
+ "AutoModel": "GPT2LMHeadModel",
8
+ "AutoModelForCausalLM": "GPT2LMHeadModel",
9
+ "AutoModelForQuestionAnswering": "GPT2LMHeadModel"
10
  },
11
  "batch_size": 12,
12
  "block_size": 100,