Toasteror commited on
Commit
7ff3ff3
·
1 Parent(s): 596a32e

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +2 -0
config.json CHANGED
@@ -1,5 +1,7 @@
1
  {
2
  "model_type": "gpt2",
 
 
3
  "vocab_size": 1000,
4
  "hidden_size": 16,
5
  "embedding_size": 16,
 
1
  {
2
  "model_type": "gpt2",
3
+ "architectures": ["GPT2LMHeadModel"],
4
+ "tokenizer_class": "GPT2Tokenizer",
5
  "vocab_size": 1000,
6
  "hidden_size": 16,
7
  "embedding_size": 16,