mwirth7 commited on
Commit
908fdb1
·
verified ·
1 Parent(s): 65b7d0b

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +2 -1
config.json CHANGED
@@ -5,7 +5,8 @@
5
  "attn_drop_rate": 0.0,
6
  "auto_map": {
7
  "AutoConfig": "configuration_bird_mae.BirdMAEConfig",
8
- "AutoModel": "modeling_bird_mae.BirdMAEModel"
 
9
  },
10
  "depth": 24,
11
  "drop_path_rate": 0.0,
 
5
  "attn_drop_rate": 0.0,
6
  "auto_map": {
7
  "AutoConfig": "configuration_bird_mae.BirdMAEConfig",
8
+ "AutoModel": "modeling_bird_mae.BirdMAEModel",
9
+ "AutoModelForSequenceClassification": "modeling_bird_mae.BirdMAEForSequenceClassification"
10
  },
11
  "depth": 24,
12
  "drop_path_rate": 0.0,