lewtun HF staff commited on
Commit
c2ec213
1 Parent(s): 87c4c0d

Register all auto classes

Browse files

This PR enables the loading of the model with all `AutoModelForXxx` classes supported in `modeling_RW.py`

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,