lewtun HF staff commited on
Commit
50a1725
1 Parent(s): eddb07e

Enable loading with auto classes

Browse files

This PR enables the model to be loaded 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,