yangwang825 commited on
Commit
93ce628
·
verified ·
1 Parent(s): a6aed96

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +3 -1
config.json CHANGED
@@ -8,7 +8,9 @@
8
  "attention_dropout": 0.1,
9
  "auto_map": {
10
  "AutoConfig": "configuration_hubert_spkreg.HubertSpkRegConfig",
11
- "AutoModel": "modeling_hubert_spkreg.HubertSpkRegModel"
 
 
12
  },
13
  "bos_token_id": 1,
14
  "classifier_proj_size": 256,
 
8
  "attention_dropout": 0.1,
9
  "auto_map": {
10
  "AutoConfig": "configuration_hubert_spkreg.HubertSpkRegConfig",
11
+ "AutoFeatureExtractor": "feature_extractor_hubert_spkreg.HubertSpkRegFeatureExtractor",
12
+ "AutoModel": "modeling_hubert_spkreg.HubertSpkRegModel",
13
+ "AutoModelForAudioClassification": "modeling_hubert_spkreg.HubertSpkRegForSequenceClassification"
14
  },
15
  "bos_token_id": 1,
16
  "classifier_proj_size": 256,