confit
/

whisper-small-spkreg

Feature Extraction

Model card Files Files and versions Community

yangwang825 commited on Nov 21, 2024

Commit

71543c5

·

verified ·

1 Parent(s): 52168ac

Upload model

Files changed (1) hide show

config.json +3 -2

config.json CHANGED Viewed

@@ -4,11 +4,12 @@
   "activation_function": "gelu",
   "apply_spec_augment": false,
   "architectures": [
-    "WhisperForConditionalGeneration"
   ],
   "attention_dropout": 0.0,
   "auto_map": {
-    "AutoConfig": "configuration_whisper_spkreg.WhisperSpkRegConfig"
   },
   "begin_suppress_tokens": [
     220,

   "activation_function": "gelu",
   "apply_spec_augment": false,
   "architectures": [
+    "WhisperSpkRegModel"
   ],
   "attention_dropout": 0.0,
   "auto_map": {
+    "AutoConfig": "configuration_whisper_spkreg.WhisperSpkRegConfig",
+    "AutoModel": "modeling_whisper_spkreg.WhisperSpkRegModel"
   },
   "begin_suppress_tokens": [
     220,