Karthik15 commited on
Commit
e406b27
·
verified ·
1 Parent(s): 05388b3

Update config.js

Browse files
Files changed (1) hide show
  1. config.js +11 -10
config.js CHANGED
@@ -1,10 +1,11 @@
1
- {
2
- "vocab_size": 50257,
3
- "block_size": 128,
4
- "n_layer": 6,
5
- "n_head": 6,
6
- "n_embd": 384,
7
- "dropout": 0.1,
8
- "bias": true,
9
- "_name_or_path": "custom-gpt"
10
- }
 
 
1
+ {
2
+ "model_type":"gpt2",
3
+ "vocab_size": 50257,
4
+ "block_size": 128,
5
+ "n_layer": 6,
6
+ "n_head": 6,
7
+ "n_embd": 384,
8
+ "dropout": 0.1,
9
+ "bias": true,
10
+ "_name_or_path": "custom-gpt"
11
+ }