Update config.json
Browse filesDue to change in the names of original files. Error is popping.
Please update the config file as modified by me( New config file is tried and tested and its Working!!!)
For time being anyone using this code use the following code while loading:
config = AutoConfig.from_pretrained('/content/drive/Shareddrives/shared drive/config.json')
model = AutoModel.from_pretrained('ybelkada/falcon-7b-sharded-bf16', config=config)
- config.json +7 -6
config.json
CHANGED
@@ -7,13 +7,14 @@
|
|
7 |
],
|
8 |
"attention_dropout": 0.0,
|
9 |
"auto_map": {
|
10 |
-
"AutoConfig": "tiiuae/falcon-7b--
|
11 |
-
"AutoModel": "tiiuae/falcon-7b--
|
12 |
-
"AutoModelForCausalLM": "tiiuae/falcon-7b--
|
13 |
-
"AutoModelForQuestionAnswering": "tiiuae/falcon-7b--
|
14 |
-
"AutoModelForSequenceClassification": "tiiuae/falcon-7b--
|
15 |
-
"AutoModelForTokenClassification": "tiiuae/falcon-7b--
|
16 |
},
|
|
|
17 |
"bias": false,
|
18 |
"bos_token_id": 11,
|
19 |
"eos_token_id": 11,
|
|
|
7 |
],
|
8 |
"attention_dropout": 0.0,
|
9 |
"auto_map": {
|
10 |
+
"AutoConfig": "tiiuae/falcon-7b--configuration_falcon.FalconConfig",
|
11 |
+
"AutoModel": "tiiuae/falcon-7b--modeling_falcon.FalconModel",
|
12 |
+
"AutoModelForCausalLM": "tiiuae/falcon-7b--modeling_falcon.FalconForCausalLM",
|
13 |
+
"AutoModelForQuestionAnswering": "tiiuae/falcon-7b--modeling_falcon.FalconForQuestionAnswering",
|
14 |
+
"AutoModelForSequenceClassification": "tiiuae/falcon-7b--modeling_falcon.FalconForSequenceClassification",
|
15 |
+
"AutoModelForTokenClassification": "tiiuae/falcon-7b--modeling_falcon.FalconForTokenClassification"
|
16 |
},
|
17 |
+
|
18 |
"bias": false,
|
19 |
"bos_token_id": 11,
|
20 |
"eos_token_id": 11,
|