BramVanroy commited on
Commit
a197e75
1 Parent(s): a79f5b0

Update config.json for newer transformers

Browse files

Instead of using custom code from Falcon (which does not work anymore), use ubilt-in transformers model code

Files changed (1) hide show
  1. config.json +7 -7
config.json CHANGED
@@ -3,16 +3,16 @@
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--configuration_RW.RWConfig",
11
- "AutoModel": "tiiuae/falcon-40b--modelling_RW.RWModel",
12
- "AutoModelForCausalLM": "tiiuae/falcon-40b--modelling_RW.RWForCausalLM",
13
- "AutoModelForQuestionAnswering": "tiiuae/falcon-40b--modelling_RW.RWForQuestionAnswering",
14
- "AutoModelForSequenceClassification": "tiiuae/falcon-40b--modelling_RW.RWForSequenceClassification",
15
- "AutoModelForTokenClassification": "tiiuae/falcon-40b--modelling_RW.RWForTokenClassification"
16
  },
17
  "bias": false,
18
  "bos_token_id": 11,
 
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": 11,