FalconLLM lewtun HF staff commited on
Commit
2e35869
1 Parent(s): 87c4c0d

Register all auto classes (#12)

Browse files

- Register all auto classes (c2ec2135bb8e04c4de57fff5cb7f3996bf70c62c)


Co-authored-by: Lewis Tunstall <lewtun@users.noreply.huggingface.co>

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,