tiiuae/falcon-40b
tiiuae/falcon-40b
1
0
Overview Code Discussions Pull Requests Wiki Activity

Enable loading with auto classes (#23)

Browse Source 
- Enable loading with auto classes (50a17258bb967c1c272d54dd9920cb4c249947f9)


Co-authored-by: Lewis Tunstall <lewtun@users.noreply.huggingface.co>
This commit is contained in:
Falcon LLM TII UAE 2023-05-31 13:16:00 +00:00 committed by system
parent eddb07e1aa
commit f431193eae
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
config.json
View File

@ -7,6 +7,10 @@
"attention_dropout": 0.0,
"auto_map": {
"AutoConfig": "configuration_RW.RWConfig",
"AutoModel": "modelling_RW.RWModel",
"AutoModelForSequenceClassification": "modelling_RW.RWForSequenceClassification",
"AutoModelForTokenClassification": "modelling_RW.RWForTokenClassification",
"AutoModelForQuestionAnswering": "modelling_RW.RWForQuestionAnswering",
"AutoModelForCausalLM": "modelling_RW.RWForCausalLM"
},
"bias": false,