yangwang825 commited on
Commit
c1841c7
·
1 Parent(s): 21ab90e

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +3 -2
config.json CHANGED
@@ -1,5 +1,5 @@
1
  {
2
- "_name_or_path": "m-a-p/MERT-v1-95M",
3
  "activation_dropout": 0.0,
4
  "apply_spec_augment": true,
5
  "architectures": [
@@ -9,7 +9,8 @@
9
  "attention_relax": -1.0,
10
  "auto_map": {
11
  "AutoConfig": "configuration_mert.MERTConfig",
12
- "AutoModel": "modeling_MERT.MERTModel"
 
13
  },
14
  "bos_token_id": 1,
15
  "classifier_proj_size": 256,
 
1
  {
2
+ "_name_or_path": "yangwang825/mert-base",
3
  "activation_dropout": 0.0,
4
  "apply_spec_augment": true,
5
  "architectures": [
 
9
  "attention_relax": -1.0,
10
  "auto_map": {
11
  "AutoConfig": "configuration_mert.MERTConfig",
12
+ "AutoModel": "modeling_mert.MERTModel",
13
+ "AutoModelForAudioClassification": "modeling_mert.MERTForSequenceClassification",
14
  },
15
  "bos_token_id": 1,
16
  "classifier_proj_size": 256,