Wonder-Griffin commited on
Commit
10a2301
·
verified ·
1 Parent(s): f129cd7

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +1 -1
config.json CHANGED
@@ -26,7 +26,7 @@
26
  },
27
  "learning_rate": 5e-05,
28
  "max_len": 512,
29
- "model_type": "GPT2LMHeadModel",
30
  "n_head": 12,
31
  "n_layer": 12,
32
  "torch_dtype": "float32",
 
26
  },
27
  "learning_rate": 5e-05,
28
  "max_len": 512,
29
+ "model_type": "gpt2",
30
  "n_head": 12,
31
  "n_layer": 12,
32
  "torch_dtype": "float32",