Update huggingface_wav2vec.py

alisafaya · web-flow · commit caa5c75b1ef1 · 2022-02-15T18:06:41.000+03:00
diff --git a/speechbrain/lobes/models/huggingface_wav2vec.py b/speechbrain/lobes/models/huggingface_wav2vec.py
@@ -39,9 +39,17 @@
 
 logger = logging.getLogger(__name__)
 
-HF_models = {"wav2vec2": Wav2Vec2Model, "hubert": HubertModel, "wavlm": WavLMModel}
-
-HF_config = {"wav2vec2": Wav2Vec2Config, "hubert": HubertConfig, "wavlm": WavLMConfig}
+HF_models = {
+    "wav2vec2": Wav2Vec2Model,
+    "hubert": HubertModel,
+    "wavlm": WavLMModel,
+}
+
+HF_config = {
+    "wav2vec2": Wav2Vec2Config,
+    "hubert": HubertConfig,
+    "wavlm": WavLMConfig,
+}
 
 
 class HuggingFaceWav2Vec2(nn.Module):
