Enabled the AutoModelForSequenceClassification in the auto_map

#22
Files changed (1) hide show
  1. config.json +1 -0
config.json CHANGED
@@ -7,6 +7,7 @@
7
  "auto_map": {
8
  "AutoConfig": "configuration_phi3_small.Phi3SmallConfig",
9
  "AutoModelForCausalLM": "modeling_phi3_small.Phi3SmallForCausalLM",
 
10
  "AutoTokenizer": "tokenization_phi3_small.Phi3SmallTokenizer"
11
  },
12
  "blocksparse_block_size": 64,
 
7
  "auto_map": {
8
  "AutoConfig": "configuration_phi3_small.Phi3SmallConfig",
9
  "AutoModelForCausalLM": "modeling_phi3_small.Phi3SmallForCausalLM",
10
+ "AutoModelForSequenceClassification": "modeling_phi3_small.Phi3SmallForSequenceClassification",
11
  "AutoTokenizer": "tokenization_phi3_small.Phi3SmallTokenizer"
12
  },
13
  "blocksparse_block_size": 64,