Rocketknight1
/

falcon-rw-1b

Text Generation

text-generation-inference

Inference Endpoints

Model card Files Files and versions Community

Rocketknight1 HF staff commited on Jul 6, 2023

Commit

8acd36b

·

1 Parent(s): e7cf6b5

Update config.json

Files changed (1) hide show

config.json +3 -3

config.json CHANGED Viewed

@@ -15,9 +15,9 @@
   "layer_norm_epsilon": 1e-05,
   "model_type": "falcon",
   "multi_query": false,
-  "n_head": 32,
-  "n_head_kv": 32,
-  "n_layer": 24,
   "new_decoder_architecture": false,
   "parallel_attn": false,
   "torch_dtype": "bfloat16",

   "layer_norm_epsilon": 1e-05,
   "model_type": "falcon",
   "multi_query": false,
+  "num_attention_heads": 32,
+  "num_kv_heads": 32,
+  "num_hidden_layers": 24,
   "new_decoder_architecture": false,
   "parallel_attn": false,
   "torch_dtype": "bfloat16",