joanllop commited on
Commit
782aab3
1 Parent(s): 203f87a

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +10 -10
config.json CHANGED
@@ -1,18 +1,18 @@
1
  {
2
- "_name_or_path": "/bscdata/models/falcon_7b_balanced_tokenizer_fp16/",
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": "configuration_falcon.FalconConfig",
11
- "AutoModel": "modeling_falcon.FalconModel",
12
- "AutoModelForSequenceClassification": "modeling_falcon.FalconForSequenceClassification",
13
- "AutoModelForTokenClassification": "modeling_falcon.FalconForTokenClassification",
14
- "AutoModelForQuestionAnswering": "modeling_falcon.FalconForQuestionAnswering",
15
- "AutoModelForCausalLM": "modeling_falcon.FalconForCausalLM"
16
  },
17
  "bias": false,
18
  "bos_token_id": 50256,
@@ -21,7 +21,7 @@
21
  "hidden_size": 4544,
22
  "initializer_range": 0.02,
23
  "layer_norm_epsilon": 1e-05,
24
- "model_type": "falcon",
25
  "multi_query": true,
26
  "n_head": 71,
27
  "n_layer": 32,
@@ -32,4 +32,4 @@
32
  "use_cache": true,
33
  "max_position_embeddings": 2048,
34
  "vocab_size": 50257
35
- }
 
1
  {
2
+ "_name_or_path": ".",
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": "configuration_RW.RWConfig",
11
+ "AutoModel": "modelling_RW.RWModel",
12
+ "AutoModelForCausalLM": "modelling_RW.RWForCausalLM",
13
+ "AutoModelForQuestionAnswering": "modelling_RW.RWForQuestionAnswering",
14
+ "AutoModelForSequenceClassification": "modelling_RW.RWForSequenceClassification",
15
+ "AutoModelForTokenClassification": "modelling_RW.RWForTokenClassification"
16
  },
17
  "bias": false,
18
  "bos_token_id": 50256,
 
21
  "hidden_size": 4544,
22
  "initializer_range": 0.02,
23
  "layer_norm_epsilon": 1e-05,
24
+ "model_type": "RefinedWebModel",
25
  "multi_query": true,
26
  "n_head": 71,
27
  "n_layer": 32,
 
32
  "use_cache": true,
33
  "max_position_embeddings": 2048,
34
  "vocab_size": 50257
35
+ }