avnishkr commited on
Commit
d5b1379
1 Parent(s): 318c118

Update config.json

Browse files

Due 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)

Files changed (1) hide show
  1. 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--configuration_RW.RWConfig",
11
- "AutoModel": "tiiuae/falcon-7b--modelling_RW.RWModel",
12
- "AutoModelForCausalLM": "tiiuae/falcon-7b--modelling_RW.RWForCausalLM",
13
- "AutoModelForQuestionAnswering": "tiiuae/falcon-7b--modelling_RW.RWForQuestionAnswering",
14
- "AutoModelForSequenceClassification": "tiiuae/falcon-7b--modelling_RW.RWForSequenceClassification",
15
- "AutoModelForTokenClassification": "tiiuae/falcon-7b--modelling_RW.RWForTokenClassification"
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,