BramVanroy commited on
Commit
b50ea88
1 Parent(s): fe535cd

Fix parameter names

Browse files
Files changed (1) hide show
  1. config.json +4 -3
config.json CHANGED
@@ -22,9 +22,10 @@
22
  "initializer_range": 0.02,
23
  "layer_norm_epsilon": 1e-05,
24
  "model_type": "falcon",
25
- "n_head": 128,
26
- "n_head_kv": 8,
27
- "n_layer": 60,
 
28
  "parallel_attn": true,
29
  "torch_dtype": "bfloat16",
30
  "transformers_version": "4.30.1",
 
22
  "initializer_range": 0.02,
23
  "layer_norm_epsilon": 1e-05,
24
  "model_type": "falcon",
25
+ "new_decoder_architecture": true,
26
+ "num_attention_heads": 128,
27
+ "num_hidden_layers": 60,
28
+ "num_kv_heads": 8,
29
  "parallel_attn": true,
30
  "torch_dtype": "bfloat16",
31
  "transformers_version": "4.30.1",