confit
/

whisper-base-spkreg

Feature Extraction

Model card Files Files and versions Community

yangwang825 commited on Nov 21, 2024

Commit

9a87507

·

verified ·

1 Parent(s): 214b742

Update config.json

Files changed (1) hide show

config.json +4 -2

config.json CHANGED Viewed

@@ -6,10 +6,12 @@
   "architectures": [
     "WhisperSpkRegModel"
   ],
-  "attention_dropout": 0.0,
   "auto_map": {
     "AutoConfig": "configuration_whisper_spkreg.WhisperSpkRegConfig",
-    "AutoModel": "modeling_whisper_spkreg.WhisperSpkRegModel"
   },
   "begin_suppress_tokens": [
     220,

   "architectures": [
     "WhisperSpkRegModel"
   ],
+  "attention_dropout": 0.1,
   "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,