Fix Typo in Argument of FlaxWav2Vec2ForPreTrainingModule (#16084)

aa4c0a86 · Sanchit Gandhi · GitHub · e231c729 · aa4c0a86
Unverified Commit aa4c0a86 authored Mar 25, 2022 by Sanchit Gandhi Committed by GitHub Mar 25, 2022
Show whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

src/transformers/models/wav2vec2/modeling_flax_wav2vec2.py src/transformers/models/wav2vec2/modeling_flax_wav2vec2.py +2 -2

No files found.
--- a/src/transformers/models/wav2vec2/modeling_flax_wav2vec2.py
+++ b/src/transformers/models/wav2vec2/modeling_flax_wav2vec2.py
@@ -1240,7 +1240,7 @@ class FlaxWav2Vec2ForPreTrainingModule(nn.Module):
        deterministic: bool = True,
        output_attentions=None,
        output_hidden_states=None,
-        freeze_feature_enocder=False,
+        freeze_feature_encoder=False,
        return_dict=None,
    ):
        r"""
@@ -1261,7 +1261,7 @@ class FlaxWav2Vec2ForPreTrainingModule(nn.Module):
            output_hidden_states=output_hidden_states,
            mask_time_indices=mask_time_indices,
            deterministic=deterministic,
-            freeze_feature_encoder=freeze_feature_enocder,
+            freeze_feature_encoder=freeze_feature_encoder,
            return_dict=return_dict,
        )