glamprou commited on
Commit
0b38e9d
·
verified ·
1 Parent(s): 25f90bf

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +1 -1
config.json CHANGED
@@ -5,7 +5,7 @@
5
  "SwitchTransformerForSequenceClassification"
6
  ],
7
  "auto_map": {
8
- "AutoModelForSequenceClassification": "SwitchTransformerForSequenceClassification"
9
  },
10
  "batch_prioritized_routing": false,
11
  "d_ff": 3072,
 
5
  "SwitchTransformerForSequenceClassification"
6
  ],
7
  "auto_map": {
8
+ "AutoModelForSequenceClassification": "modeling_switch_transformers.SwitchTransformerForSequenceClassification"
9
  },
10
  "batch_prioritized_routing": false,
11
  "d_ff": 3072,