yizhilll commited on
Commit
7a16d1f
1 Parent(s): 393a70b

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +6 -2
config.json CHANGED
@@ -1,10 +1,15 @@
1
  {
2
- "_name_or_path": "m-a-p/MERT-v1",
3
  "activation_dropout": 0.1,
4
  "apply_spec_augment": true,
5
  "architectures": [
6
  "MERTModel"
7
  ],
 
 
 
 
 
8
  "sample_rate": 16000,
9
  "attention_dropout": 0.1,
10
  "bos_token_id": 1,
@@ -62,7 +67,6 @@
62
  "mask_time_length": 10,
63
  "mask_time_min_masks": 2,
64
  "mask_time_prob": 0.05,
65
- "model_type": "mert_model",
66
  "num_attention_heads": 12,
67
  "num_conv_pos_embedding_groups": 16,
68
  "num_conv_pos_embeddings": 128,
 
1
  {
2
+ "_name_or_path": "m-a-p/MERT-v0",
3
  "activation_dropout": 0.1,
4
  "apply_spec_augment": true,
5
  "architectures": [
6
  "MERTModel"
7
  ],
8
+ "model_type": "mert_model",
9
+ "auto_map": {
10
+ "AutoConfig": "configuration_MERT.MERTConfig",
11
+ "AutoModel": "modeling_MERT.MERTModel"
12
+ },
13
  "sample_rate": 16000,
14
  "attention_dropout": 0.1,
15
  "bos_token_id": 1,
 
67
  "mask_time_length": 10,
68
  "mask_time_min_masks": 2,
69
  "mask_time_prob": 0.05,
 
70
  "num_attention_heads": 12,
71
  "num_conv_pos_embedding_groups": 16,
72
  "num_conv_pos_embeddings": 128,