FalconLLM lewtun HF staff commited on
Commit
f431193
1 Parent(s): eddb07e

Enable loading with auto classes (#23)

Browse files

- Enable loading with auto classes (50a17258bb967c1c272d54dd9920cb4c249947f9)


Co-authored-by: Lewis Tunstall <[email protected]>

Files changed (1) hide show
  1. config.json +4 -0
config.json CHANGED
@@ -7,6 +7,10 @@
7
  "attention_dropout": 0.0,
8
  "auto_map": {
9
  "AutoConfig": "configuration_RW.RWConfig",
 
 
 
 
10
  "AutoModelForCausalLM": "modelling_RW.RWForCausalLM"
11
  },
12
  "bias": false,
 
7
  "attention_dropout": 0.0,
8
  "auto_map": {
9
  "AutoConfig": "configuration_RW.RWConfig",
10
+ "AutoModel": "modelling_RW.RWModel",
11
+ "AutoModelForSequenceClassification": "modelling_RW.RWForSequenceClassification",
12
+ "AutoModelForTokenClassification": "modelling_RW.RWForTokenClassification",
13
+ "AutoModelForQuestionAnswering": "modelling_RW.RWForQuestionAnswering",
14
  "AutoModelForCausalLM": "modelling_RW.RWForCausalLM"
15
  },
16
  "bias": false,