DongfuJiang commited on
Commit
d9d09a3
·
verified ·
1 Parent(s): ac923ff

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +1 -0
config.json CHANGED
@@ -32,6 +32,7 @@
32
  "num_hidden_layers": 28,
33
  "num_key_value_heads": 4,
34
  "problem_type": "regression",
 
35
  "rms_norm_eps": 1e-06,
36
  "rope_scaling": {
37
  "mrope_section": [
 
32
  "num_hidden_layers": 28,
33
  "num_key_value_heads": 4,
34
  "problem_type": "regression",
35
+ "num_labels": 5,
36
  "rms_norm_eps": 1e-06,
37
  "rope_scaling": {
38
  "mrope_section": [