michaelbzhu commited on
Commit
3284ef7
·
verified ·
1 Parent(s): be4cbb2

add model_type to config

Browse files
Files changed (1) hide show
  1. config.json +2 -1
config.json CHANGED
@@ -3,5 +3,6 @@
3
  "d_model": 8192,
4
  "n_heads": 64,
5
  "n_layers": 3,
6
- "n_vocab": 50257
 
7
  }
 
3
  "d_model": 8192,
4
  "n_heads": 64,
5
  "n_layers": 3,
6
+ "n_vocab": 50257,
7
+ "model_type": "gpt2"
8
  }