confit
/

wavlm-base-spkreg

Feature Extraction

Model card Files Files and versions Community

yangwang825 commited on Nov 21, 2024

Commit

44bea38

·

verified ·

1 Parent(s): 8d8de54

Update config.json

Files changed (1) hide show

config.json +3 -1

config.json CHANGED Viewed

@@ -11,7 +11,9 @@
   "attention_dropout": 0.1,
   "auto_map": {
     "AutoConfig": "configuration_wavlm_spkreg.WavLMSpkRegConfig",
-    "AutoModel": "modeling_wavlm_spkreg.WavLMSpkRegModel"
   },
   "bos_token_id": 1,
   "classifier_proj_size": 256,

   "attention_dropout": 0.1,
   "auto_map": {
     "AutoConfig": "configuration_wavlm_spkreg.WavLMSpkRegConfig",
+    "AutoFeatureExtractor": "feature_extractor_wavlm_spkreg.WavLMSpkRegFeatureExtractor",
+    "AutoModel": "modeling_wavlm_spkreg.WavLMSpkRegModel",
+    "AutoModelForAudioClassification": "modeling_wavlm_spkreg.WavLMSpkRegForSequenceClassification"
   },
   "bos_token_id": 1,
   "classifier_proj_size": 256,