bapatra mrm196 commited on
Commit
69caae1
1 Parent(s): f196467

Enabled the AutoModelForSequenceClassification in the auto_map (#22)

Browse files

- Enabled the AutoModelForSequenceClassification in the auto_map (32e806687563ec1bc38f25f5605b72ed83dfaed5)


Co-authored-by: Mohammadreza Mohseni <[email protected]>

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,