confit
/

wav2vec2-base-spkreg

Feature Extraction

wav2vec2_spkreg

Model card Files Files and versions Community

yangwang825 commited on Nov 21

Commit

49950e3

•

1 Parent(s): e2280a5

Update config.json

Files changed (1) hide show

config.json +3 -1

config.json CHANGED Viewed

@@ -12,7 +12,9 @@
   "attention_dropout": 0.1,
   "auto_map": {
     "AutoConfig": "configuration_wav2vec2_spkreg.Wav2Vec2SpkRegConfig",
-    "AutoModel": "modeling_wav2vec2_spkreg.Wav2Vec2SpkRegModel"
   },
   "bos_token_id": 1,
   "classifier_proj_size": 256,

   "attention_dropout": 0.1,
   "auto_map": {
     "AutoConfig": "configuration_wav2vec2_spkreg.Wav2Vec2SpkRegConfig",
+    "AutoFeatureExtractor": "feature_extractor_wav2vec2_spkreg.Wav2Vec2SpkRegFeatureExtractor",
+    "AutoModel": "modeling_wav2vec2_spkreg.Wav2Vec2SpkRegModel",
+    "AutoModelForAudioClassification": "modeling_wav2vec2_spkreg.Wav2Vec2SpkRegForSequenceClassification"
   },
   "bos_token_id": 1,
   "classifier_proj_size": 256,