confit
/

whisper-base-spkreg

Feature Extraction

Model card Files Files and versions

yangwang825 commited on Nov 21, 2024

Commit

3e00a90

·

verified ·

1 Parent(s): 28c8783

Update config.json

Files changed (1) hide show

config.json +3 -1

config.json CHANGED Viewed

@@ -9,7 +9,9 @@
   "attention_dropout": 0.0,
   "auto_map": {
     "AutoConfig": "configuration_whisper_spkreg.WhisperSpkRegConfig",
-    "AutoModel": "modeling_whisper_spkreg.WhisperSpkRegModel"
   },
   "begin_suppress_tokens": [
     220,

   "attention_dropout": 0.0,
   "auto_map": {
     "AutoConfig": "configuration_whisper_spkreg.WhisperSpkRegConfig",
+    "AutoFeatureExtractor": "feature_extractor_whisper_spkreg.WhisperSpkRegFeatureExtractor",
+    "AutoModel": "modeling_whisper_spkreg.WhisperSpkRegModel",
+    "AutoModelForAudioClassification": "modeling_whisper_spkreg.WhisperSpkRegForSequenceClassification"
   },
   "begin_suppress_tokens": [
     220,