SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

[Wav2Vec2FeatureExtractor] smal fixes (#10455)

Browse Source 
* smal fixes

* don't check for None
This commit is contained in:
Suraj Patil
2021-03-01 20:19:52 +05:30
committed by GitHub
parent 11655fafdd
commit a106bde5a7
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
src/transformers/models/wav2vec2/feature_extraction_wav2vec2.py
View File

@@ -47,7 +47,7 @@ class Wav2Vec2FeatureExtractor(PreTrainedFeatureExtractor):
improve the performance for some models, *e.g.*, `wav2vec2-lv60 improve the performance for some models, *e.g.*, `wav2vec2-lv60
<https://huggingface.co/models?search=lv60>`__. <https://huggingface.co/models?search=lv60>`__.
return_attention_mask (:obj:`bool`, `optional`, defaults to :obj:`False`): 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:: .. note::
@@ -89,6 +89,7 @@ class Wav2Vec2FeatureExtractor(PreTrainedFeatureExtractor):
padding: Union[bool, str, PaddingStrategy] = False, padding: Union[bool, str, PaddingStrategy] = False,
max_length: Optional[int] = None, max_length: Optional[int] = None,
pad_to_multiple_of: Optional[int] = None, pad_to_multiple_of: Optional[int] = None,
return_attention_mask: Optional[bool] = None,
return_tensors: Optional[Union[str, TensorType]] = None, return_tensors: Optional[Union[str, TensorType]] = None,
sampling_rate: Optional[int] = None, sampling_rate: Optional[int] = None,
**kwargs **kwargs
@@ -185,7 +186,7 @@ class Wav2Vec2FeatureExtractor(PreTrainedFeatureExtractor):
padding=padding, padding=padding,
max_length=max_length, max_length=max_length,
pad_to_multiple_of=pad_to_multiple_of, 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, return_tensors=return_tensors,
) )