mrm196 commited on
Commit
a53528a
1 Parent(s): 5be6479

Add the classifiers to the auto_map

Browse files

Added the following classifiers to the `auto_map` in the `config.json` to enable their discovery:
- `AutoModelForSequenceClassification`
- `AutoModelForTokenClassification`

Files changed (1) hide show
  1. config.json +3 -1
config.json CHANGED
@@ -6,7 +6,9 @@
6
  "attention_dropout": 0.0,
7
  "auto_map": {
8
  "AutoConfig": "configuration_phi3.Phi3Config",
9
- "AutoModelForCausalLM": "modeling_phi3.Phi3ForCausalLM"
 
 
10
  },
11
  "bos_token_id": 1,
12
  "embd_pdrop": 0.0,
 
6
  "attention_dropout": 0.0,
7
  "auto_map": {
8
  "AutoConfig": "configuration_phi3.Phi3Config",
9
+ "AutoModelForCausalLM": "modeling_phi3.Phi3ForCausalLM",
10
+ "AutoModelForSequenceClassification": "modeling_phi3.Phi3ForSequenceClassification",
11
+ "AutoModelForTokenClassification": "modeling_phi3.Phi3ForTokenClassification"
12
  },
13
  "bos_token_id": 1,
14
  "embd_pdrop": 0.0,