Update config.json

#5
by lucas0 - opened
Files changed (1) hide show
  1. config.json +8 -7
config.json CHANGED
@@ -3,12 +3,12 @@
3
  "alibi": false,
4
  "apply_residual_connection_post_layernorm": false,
5
  "architectures": [
6
- "RWForCausalLM"
7
  ],
8
  "attention_dropout": 0.0,
9
  "auto_map": {
10
- "AutoConfig": "tiiuae/falcon-40b-instruct--configuration_RW.RWConfig",
11
- "AutoModelForCausalLM": "tiiuae/falcon-40b-instruct--modelling_RW.RWForCausalLM"
12
  },
13
  "bias": false,
14
  "bos_token_id": 11,
@@ -17,10 +17,11 @@
17
  "hidden_size": 8192,
18
  "initializer_range": 0.02,
19
  "layer_norm_epsilon": 1e-05,
20
- "model_type": "RefinedWeb",
21
- "n_head": 128,
22
- "n_head_kv": 8,
23
- "n_layer": 60,
 
24
  "parallel_attn": true,
25
  "quantization_config": {
26
  "bnb_4bit_compute_dtype": "float32",
 
3
  "alibi": false,
4
  "apply_residual_connection_post_layernorm": false,
5
  "architectures": [
6
+ "FalconForCausalLM"
7
  ],
8
  "attention_dropout": 0.0,
9
  "auto_map": {
10
+ "AutoConfig": "tiiuae/falcon-40b-instruct--configuration_falcon.FalconConfig",
11
+ "AutoModelForCausalLM": "tiiuae/falcon-40b-instruct--modeling_falcon.FalconModel"
12
  },
13
  "bias": false,
14
  "bos_token_id": 11,
 
17
  "hidden_size": 8192,
18
  "initializer_range": 0.02,
19
  "layer_norm_epsilon": 1e-05,
20
+ "model_type": "falcon",
21
+ "new_decoder_architecture": true,
22
+ "num_attention_heads": 128,
23
+ "num_hidden_layers": 60,
24
+ "num_kv_heads": 8,
25
  "parallel_attn": true,
26
  "quantization_config": {
27
  "bnb_4bit_compute_dtype": "float32",