From a106bde5a741878e18aa4547de2d9ab46d9a106d Mon Sep 17 00:00:00 2001 From: Suraj Patil Date: Mon, 1 Mar 2021 20:19:52 +0530 Subject: [PATCH] [Wav2Vec2FeatureExtractor] smal fixes (#10455) * smal fixes * don't check for None --- .../models/wav2vec2/feature_extraction_wav2vec2.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/src/transformers/models/wav2vec2/feature_extraction_wav2vec2.py b/src/transformers/models/wav2vec2/feature_extraction_wav2vec2.py index 9fd3f9a4c1..bc4297c1ac 100644 --- a/src/transformers/models/wav2vec2/feature_extraction_wav2vec2.py +++ b/src/transformers/models/wav2vec2/feature_extraction_wav2vec2.py @@ -47,7 +47,7 @@ class Wav2Vec2FeatureExtractor(PreTrainedFeatureExtractor): improve the performance for some models, *e.g.*, `wav2vec2-lv60 `__. return_attention_mask (:obj:`bool`, `optional`, defaults to :obj:`False`): - Whether or not :meth:`~transformers.Wav2Vec2Tokenizer.__call__` should return :obj:`attention_mask`. + Whether or not :meth:`~transformers.Wav2Vec2FeatureExtractor.__call__` should return :obj:`attention_mask`. .. note:: @@ -89,6 +89,7 @@ class Wav2Vec2FeatureExtractor(PreTrainedFeatureExtractor): padding: Union[bool, str, PaddingStrategy] = False, max_length: Optional[int] = None, pad_to_multiple_of: Optional[int] = None, + return_attention_mask: Optional[bool] = None, return_tensors: Optional[Union[str, TensorType]] = None, sampling_rate: Optional[int] = None, **kwargs @@ -185,7 +186,7 @@ class Wav2Vec2FeatureExtractor(PreTrainedFeatureExtractor): padding=padding, max_length=max_length, pad_to_multiple_of=pad_to_multiple_of, - return_attention_mask=self.return_attention_mask, + return_attention_mask=return_attention_mask, return_tensors=return_tensors, )