GenerTeam commited on
Commit
7657673
·
verified ·
1 Parent(s): 6719ee4

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +1 -0
config.json CHANGED
@@ -31,6 +31,7 @@
31
  "num_attention_heads": 16,
32
  "num_hidden_layers": 28,
33
  "num_key_value_heads": 4,
 
34
  "pad_token_id": 3,
35
  "pretraining_tp": 1,
36
  "rms_norm_eps": 1e-05,
 
31
  "num_attention_heads": 16,
32
  "num_hidden_layers": 28,
33
  "num_key_value_heads": 4,
34
+ "num_prediction_heads": 2,
35
  "pad_token_id": 3,
36
  "pretraining_tp": 1,
37
  "rms_norm_eps": 1e-05,