Replace kv_channels with head_dim
Browse files- config.json +1 -1
config.json
CHANGED
|
@@ -13,7 +13,7 @@
|
|
| 13 |
"model_type": "nemotron",
|
| 14 |
"num_attention_heads": 48,
|
| 15 |
"num_hidden_layers": 32,
|
| 16 |
-
"
|
| 17 |
"num_key_value_heads": 8,
|
| 18 |
"norm_eps": 1e-05,
|
| 19 |
"rope_theta": 10000,
|
|
|
|
| 13 |
"model_type": "nemotron",
|
| 14 |
"num_attention_heads": 48,
|
| 15 |
"num_hidden_layers": 32,
|
| 16 |
+
"head_dim": 128,
|
| 17 |
"num_key_value_heads": 8,
|
| 18 |
"norm_eps": 1e-05,
|
| 19 |
"rope_theta": 10000,
|