Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
chenpangpang
transformers
Commits
088c1880
Unverified
Commit
088c1880
authored
Mar 25, 2022
by
Sylvain Gugger
Committed by
GitHub
Mar 25, 2022
Browse files
Big file_utils cleanup (#16396)
* Big file_utils cleanup * This one still needs to be treated separately
parent
2b23e080
Changes
222
Hide whitespace changes
Inline
Side-by-side
Showing
20 changed files
with
30 additions
and
30 deletions
+30
-30
src/transformers/models/unispeech_sat/modeling_unispeech_sat.py
...ansformers/models/unispeech_sat/modeling_unispeech_sat.py
+1
-1
src/transformers/models/van/modeling_van.py
src/transformers/models/van/modeling_van.py
+1
-1
src/transformers/models/vilt/feature_extraction_vilt.py
src/transformers/models/vilt/feature_extraction_vilt.py
+2
-2
src/transformers/models/vilt/modeling_vilt.py
src/transformers/models/vilt/modeling_vilt.py
+2
-2
src/transformers/models/vision_encoder_decoder/modeling_flax_vision_encoder_decoder.py
...n_encoder_decoder/modeling_flax_vision_encoder_decoder.py
+4
-4
src/transformers/models/vision_encoder_decoder/modeling_tf_vision_encoder_decoder.py
...ion_encoder_decoder/modeling_tf_vision_encoder_decoder.py
+1
-1
src/transformers/models/vision_encoder_decoder/modeling_vision_encoder_decoder.py
...vision_encoder_decoder/modeling_vision_encoder_decoder.py
+1
-1
src/transformers/models/vision_text_dual_encoder/modeling_flax_vision_text_dual_encoder.py
...xt_dual_encoder/modeling_flax_vision_text_dual_encoder.py
+1
-1
src/transformers/models/vision_text_dual_encoder/modeling_vision_text_dual_encoder.py
...on_text_dual_encoder/modeling_vision_text_dual_encoder.py
+3
-3
src/transformers/models/vision_text_dual_encoder/processing_vision_text_dual_encoder.py
..._text_dual_encoder/processing_vision_text_dual_encoder.py
+1
-1
src/transformers/models/visual_bert/modeling_visual_bert.py
src/transformers/models/visual_bert/modeling_visual_bert.py
+1
-1
src/transformers/models/vit/feature_extraction_vit.py
src/transformers/models/vit/feature_extraction_vit.py
+1
-1
src/transformers/models/vit/modeling_flax_vit.py
src/transformers/models/vit/modeling_flax_vit.py
+1
-1
src/transformers/models/vit/modeling_tf_vit.py
src/transformers/models/vit/modeling_tf_vit.py
+2
-2
src/transformers/models/vit/modeling_vit.py
src/transformers/models/vit/modeling_vit.py
+1
-1
src/transformers/models/vit_mae/modeling_vit_mae.py
src/transformers/models/vit_mae/modeling_vit_mae.py
+1
-1
src/transformers/models/wav2vec2/feature_extraction_wav2vec2.py
...ansformers/models/wav2vec2/feature_extraction_wav2vec2.py
+2
-2
src/transformers/models/wav2vec2/modeling_flax_wav2vec2.py
src/transformers/models/wav2vec2/modeling_flax_wav2vec2.py
+1
-1
src/transformers/models/wav2vec2/modeling_tf_wav2vec2.py
src/transformers/models/wav2vec2/modeling_tf_wav2vec2.py
+2
-2
src/transformers/models/wav2vec2/modeling_wav2vec2.py
src/transformers/models/wav2vec2/modeling_wav2vec2.py
+1
-1
No files found.
src/transformers/models/unispeech_sat/modeling_unispeech_sat.py
View file @
088c1880
...
@@ -1116,7 +1116,7 @@ UNISPEECH_SAT_INPUTS_DOCSTRING = r"""
...
@@ -1116,7 +1116,7 @@ UNISPEECH_SAT_INPUTS_DOCSTRING = r"""
Whether or not to return the hidden states of all layers. See `hidden_states` under returned tensors for
Whether or not to return the hidden states of all layers. See `hidden_states` under returned tensors for
more detail.
more detail.
return_dict (`bool`, *optional*):
return_dict (`bool`, *optional*):
Whether or not to return a [`~
file_
utils.ModelOutput`] instead of a plain tuple.
Whether or not to return a [`~utils.ModelOutput`] instead of a plain tuple.
"""
"""
...
...
src/transformers/models/van/modeling_van.py
View file @
088c1880
...
@@ -444,7 +444,7 @@ VAN_INPUTS_DOCSTRING = r"""
...
@@ -444,7 +444,7 @@ VAN_INPUTS_DOCSTRING = r"""
Whether or not to return the hidden states of all stages. See `hidden_states` under returned tensors for
Whether or not to return the hidden states of all stages. See `hidden_states` under returned tensors for
more detail.
more detail.
return_dict (`bool`, *optional*):
return_dict (`bool`, *optional*):
Whether or not to return a [`~
file_
utils.ModelOutput`] instead of a plain tuple.
Whether or not to return a [`~utils.ModelOutput`] instead of a plain tuple.
"""
"""
...
...
src/transformers/models/vilt/feature_extraction_vilt.py
View file @
088c1880
...
@@ -145,7 +145,7 @@ class ViltFeatureExtractor(FeatureExtractionMixin, ImageFeatureExtractionMixin):
...
@@ -145,7 +145,7 @@ class ViltFeatureExtractor(FeatureExtractionMixin, ImageFeatureExtractionMixin):
Args:
Args:
pixel_values_list (`List[torch.Tensor]`):
pixel_values_list (`List[torch.Tensor]`):
List of images (pixel values) to be padded. Each image should be a tensor of shape (C, H, W).
List of images (pixel values) to be padded. Each image should be a tensor of shape (C, H, W).
return_tensors (`str` or [`~
file_
utils.TensorType`], *optional*):
return_tensors (`str` or [`~utils.TensorType`], *optional*):
If set, will return tensors instead of NumPy arrays. If set to `'pt'`, return PyTorch `torch.Tensor`
If set, will return tensors instead of NumPy arrays. If set to `'pt'`, return PyTorch `torch.Tensor`
objects.
objects.
...
@@ -208,7 +208,7 @@ class ViltFeatureExtractor(FeatureExtractionMixin, ImageFeatureExtractionMixin):
...
@@ -208,7 +208,7 @@ class ViltFeatureExtractor(FeatureExtractionMixin, ImageFeatureExtractionMixin):
- 1 for pixels that are real (i.e. **not masked**),
- 1 for pixels that are real (i.e. **not masked**),
- 0 for pixels that are padding (i.e. **masked**).
- 0 for pixels that are padding (i.e. **masked**).
return_tensors (`str` or [`~
file_
utils.TensorType`], *optional*, defaults to `'np'`):
return_tensors (`str` or [`~utils.TensorType`], *optional*, defaults to `'np'`):
If set, will return tensors of a particular framework. Acceptable values are:
If set, will return tensors of a particular framework. Acceptable values are:
- `'tf'`: Return TensorFlow `tf.constant` objects.
- `'tf'`: Return TensorFlow `tf.constant` objects.
...
...
src/transformers/models/vilt/modeling_vilt.py
View file @
088c1880
...
@@ -660,7 +660,7 @@ VILT_INPUTS_DOCSTRING = r"""
...
@@ -660,7 +660,7 @@ VILT_INPUTS_DOCSTRING = r"""
Whether or not to return the hidden states of all layers. See `hidden_states` under returned tensors for
Whether or not to return the hidden states of all layers. See `hidden_states` under returned tensors for
more detail.
more detail.
return_dict (`bool`, *optional*):
return_dict (`bool`, *optional*):
Whether or not to return a [`~
file_
utils.ModelOutput`] instead of a plain tuple.
Whether or not to return a [`~utils.ModelOutput`] instead of a plain tuple.
"""
"""
VILT_IMAGES_AND_TEXT_CLASSIFICATION_INPUTS_DOCSTRING
=
r
"""
VILT_IMAGES_AND_TEXT_CLASSIFICATION_INPUTS_DOCSTRING
=
r
"""
...
@@ -715,7 +715,7 @@ VILT_IMAGES_AND_TEXT_CLASSIFICATION_INPUTS_DOCSTRING = r"""
...
@@ -715,7 +715,7 @@ VILT_IMAGES_AND_TEXT_CLASSIFICATION_INPUTS_DOCSTRING = r"""
Whether or not to return the hidden states of all layers. See `hidden_states` under returned tensors for
Whether or not to return the hidden states of all layers. See `hidden_states` under returned tensors for
more detail.
more detail.
return_dict (`bool`, *optional*):
return_dict (`bool`, *optional*):
Whether or not to return a [`~
file_
utils.ModelOutput`] instead of a plain tuple.
Whether or not to return a [`~utils.ModelOutput`] instead of a plain tuple.
"""
"""
...
...
src/transformers/models/vision_encoder_decoder/modeling_flax_vision_encoder_decoder.py
View file @
088c1880
...
@@ -107,7 +107,7 @@ VISION_ENCODER_DECODER_INPUTS_DOCSTRING = r"""
...
@@ -107,7 +107,7 @@ VISION_ENCODER_DECODER_INPUTS_DOCSTRING = r"""
Whether or not to return the hidden states of all layers. See `hidden_states` under returned tensors for
Whether or not to return the hidden states of all layers. See `hidden_states` under returned tensors for
more detail.
more detail.
return_dict (`bool`, *optional*):
return_dict (`bool`, *optional*):
If set to `True`, the model will return a [`~
file_
utils.FlaxSeq2SeqLMOutput`] instead of a plain tuple.
If set to `True`, the model will return a [`~utils.FlaxSeq2SeqLMOutput`] instead of a plain tuple.
"""
"""
VISION_ENCODER_DECODER_ENCODE_INPUTS_DOCSTRING
=
r
"""
VISION_ENCODER_DECODER_ENCODE_INPUTS_DOCSTRING
=
r
"""
...
@@ -122,7 +122,7 @@ VISION_ENCODER_DECODER_ENCODE_INPUTS_DOCSTRING = r"""
...
@@ -122,7 +122,7 @@ VISION_ENCODER_DECODER_ENCODE_INPUTS_DOCSTRING = r"""
Whether or not to return the hidden states of all layers. See `hidden_states` under returned tensors for
Whether or not to return the hidden states of all layers. See `hidden_states` under returned tensors for
more detail.
more detail.
return_dict (`bool`, *optional*):
return_dict (`bool`, *optional*):
If set to `True`, the model will return a [`~
file_
utils.FlaxBaseModelOutput`] instead of a plain tuple.
If set to `True`, the model will return a [`~utils.FlaxBaseModelOutput`] instead of a plain tuple.
"""
"""
VISION_ENCODER_DECODER_DECODE_INPUTS_DOCSTRING
=
r
"""
VISION_ENCODER_DECODER_DECODE_INPUTS_DOCSTRING
=
r
"""
...
@@ -161,8 +161,8 @@ VISION_ENCODER_DECODER_DECODE_INPUTS_DOCSTRING = r"""
...
@@ -161,8 +161,8 @@ VISION_ENCODER_DECODER_DECODE_INPUTS_DOCSTRING = r"""
Whether or not to return the hidden states of all layers. See `hidden_states` under returned tensors for
Whether or not to return the hidden states of all layers. See `hidden_states` under returned tensors for
more detail.
more detail.
return_dict (`bool`, *optional*):
return_dict (`bool`, *optional*):
If set to `True`, the model will return a [`~
file_
utils.FlaxCausalLMOutputWithCrossAttentions`] instead of
If set to `True`, the model will return a [`~utils.FlaxCausalLMOutputWithCrossAttentions`] instead of
a
a
plain tuple.
plain tuple.
"""
"""
...
...
src/transformers/models/vision_encoder_decoder/modeling_tf_vision_encoder_decoder.py
View file @
088c1880
...
@@ -132,7 +132,7 @@ VISION_ENCODER_DECODER_INPUTS_DOCSTRING = r"""
...
@@ -132,7 +132,7 @@ VISION_ENCODER_DECODER_INPUTS_DOCSTRING = r"""
Whether or not to return the hidden states of all layers. See `hidden_states` under returned tensors for
Whether or not to return the hidden states of all layers. See `hidden_states` under returned tensors for
more detail.
more detail.
return_dict (`bool`, *optional*):
return_dict (`bool`, *optional*):
If set to `True`, the model will return a [`~
file_
utils.Seq2SeqLMOutput`] instead of a plain tuple.
If set to `True`, the model will return a [`~utils.Seq2SeqLMOutput`] instead of a plain tuple.
training (`bool`, *optional*, defaults to `False`):
training (`bool`, *optional*, defaults to `False`):
Whether or not to use the model in training mode (some modules like dropout modules have different
Whether or not to use the model in training mode (some modules like dropout modules have different
behaviors between training and evaluation).
behaviors between training and evaluation).
...
...
src/transformers/models/vision_encoder_decoder/modeling_vision_encoder_decoder.py
View file @
088c1880
...
@@ -136,7 +136,7 @@ VISION_ENCODER_DECODER_INPUTS_DOCSTRING = r"""
...
@@ -136,7 +136,7 @@ VISION_ENCODER_DECODER_INPUTS_DOCSTRING = r"""
Whether or not to return the hidden states of all layers. See `hidden_states` under returned tensors for
Whether or not to return the hidden states of all layers. See `hidden_states` under returned tensors for
more detail.
more detail.
return_dict (`bool`, *optional*):
return_dict (`bool`, *optional*):
If set to `True`, the model will return a [`~
file_
utils.Seq2SeqLMOutput`] instead of a plain tuple.
If set to `True`, the model will return a [`~utils.Seq2SeqLMOutput`] instead of a plain tuple.
kwargs: (*optional*) Remaining dictionary of keyword arguments. Keyword arguments come in two flavors:
kwargs: (*optional*) Remaining dictionary of keyword arguments. Keyword arguments come in two flavors:
- Without a prefix which will be input as `**encoder_kwargs` for the encoder forward function.
- Without a prefix which will be input as `**encoder_kwargs` for the encoder forward function.
...
...
src/transformers/models/vision_text_dual_encoder/modeling_flax_vision_text_dual_encoder.py
View file @
088c1880
...
@@ -114,7 +114,7 @@ VISION_TEXT_DUAL_ENCODER_INPUTS_DOCSTRING = r"""
...
@@ -114,7 +114,7 @@ VISION_TEXT_DUAL_ENCODER_INPUTS_DOCSTRING = r"""
Whether or not to return the hidden states of all layers. See `hidden_states` under returned tensors for
Whether or not to return the hidden states of all layers. See `hidden_states` under returned tensors for
more detail.
more detail.
return_dict (`bool`, *optional*):
return_dict (`bool`, *optional*):
Whether or not to return a [`~
file_
utils.ModelOutput`] instead of a plain tuple.
Whether or not to return a [`~utils.ModelOutput`] instead of a plain tuple.
"""
"""
...
...
src/transformers/models/vision_text_dual_encoder/modeling_vision_text_dual_encoder.py
View file @
088c1880
...
@@ -89,7 +89,7 @@ VISION_TEXT_DUAL_ENCODER_TEXT_INPUTS_DOCSTRING = r"""
...
@@ -89,7 +89,7 @@ VISION_TEXT_DUAL_ENCODER_TEXT_INPUTS_DOCSTRING = r"""
Whether or not to return the hidden states of all layers. See `hidden_states` under returned tensors for
Whether or not to return the hidden states of all layers. See `hidden_states` under returned tensors for
more detail.
more detail.
return_dict (`bool`, *optional*):
return_dict (`bool`, *optional*):
Whether or not to return a [`~
file_
utils.ModelOutput`] instead of a plain tuple.
Whether or not to return a [`~utils.ModelOutput`] instead of a plain tuple.
"""
"""
VISION_TEXT_DUAL_ENCODER_VISION_INPUTS_DOCSTRING
=
r
"""
VISION_TEXT_DUAL_ENCODER_VISION_INPUTS_DOCSTRING
=
r
"""
...
@@ -104,7 +104,7 @@ VISION_TEXT_DUAL_ENCODER_VISION_INPUTS_DOCSTRING = r"""
...
@@ -104,7 +104,7 @@ VISION_TEXT_DUAL_ENCODER_VISION_INPUTS_DOCSTRING = r"""
Whether or not to return the hidden states of all layers. See `hidden_states` under returned tensors for
Whether or not to return the hidden states of all layers. See `hidden_states` under returned tensors for
more detail.
more detail.
return_dict (`bool`, *optional*):
return_dict (`bool`, *optional*):
Whether or not to return a [`~
file_
utils.ModelOutput`] instead of a plain tuple.
Whether or not to return a [`~utils.ModelOutput`] instead of a plain tuple.
"""
"""
VISION_TEXT_DUAL_ENCODER_INPUTS_DOCSTRING
=
r
"""
VISION_TEXT_DUAL_ENCODER_INPUTS_DOCSTRING
=
r
"""
...
@@ -142,7 +142,7 @@ VISION_TEXT_DUAL_ENCODER_INPUTS_DOCSTRING = r"""
...
@@ -142,7 +142,7 @@ VISION_TEXT_DUAL_ENCODER_INPUTS_DOCSTRING = r"""
Whether or not to return the hidden states of all layers. See `hidden_states` under returned tensors for
Whether or not to return the hidden states of all layers. See `hidden_states` under returned tensors for
more detail.
more detail.
return_dict (`bool`, *optional*):
return_dict (`bool`, *optional*):
Whether or not to return a [`~
file_
utils.ModelOutput`] instead of a plain tuple.
Whether or not to return a [`~utils.ModelOutput`] instead of a plain tuple.
"""
"""
...
...
src/transformers/models/vision_text_dual_encoder/processing_vision_text_dual_encoder.py
View file @
088c1880
...
@@ -60,7 +60,7 @@ class VisionTextDualEncoderProcessor(ProcessorMixin):
...
@@ -60,7 +60,7 @@ class VisionTextDualEncoderProcessor(ProcessorMixin):
tensor. In case of a NumPy array/PyTorch tensor, each image should be of shape (C, H, W), where C is a
tensor. In case of a NumPy array/PyTorch tensor, each image should be of shape (C, H, W), where C is a
number of channels, H and W are image height and width.
number of channels, H and W are image height and width.
return_tensors (`str` or [`~
file_
utils.TensorType`], *optional*):
return_tensors (`str` or [`~utils.TensorType`], *optional*):
If set, will return tensors of a particular framework. Acceptable values are:
If set, will return tensors of a particular framework. Acceptable values are:
- `'tf'`: Return TensorFlow `tf.constant` objects.
- `'tf'`: Return TensorFlow `tf.constant` objects.
...
...
src/transformers/models/visual_bert/modeling_visual_bert.py
View file @
088c1880
...
@@ -669,7 +669,7 @@ VISUAL_BERT_INPUTS_DOCSTRING = r"""
...
@@ -669,7 +669,7 @@ VISUAL_BERT_INPUTS_DOCSTRING = r"""
Whether or not to return the hidden states of all layers. See `hidden_states` under returned tensors for
Whether or not to return the hidden states of all layers. See `hidden_states` under returned tensors for
more detail.
more detail.
return_dict (`bool`, *optional*):
return_dict (`bool`, *optional*):
Whether or not to return a [`~
file_
utils.ModelOutput`] instead of a plain tuple.
Whether or not to return a [`~utils.ModelOutput`] instead of a plain tuple.
"""
"""
...
...
src/transformers/models/vit/feature_extraction_vit.py
View file @
088c1880
...
@@ -98,7 +98,7 @@ class ViTFeatureExtractor(FeatureExtractionMixin, ImageFeatureExtractionMixin):
...
@@ -98,7 +98,7 @@ class ViTFeatureExtractor(FeatureExtractionMixin, ImageFeatureExtractionMixin):
tensor. In case of a NumPy array/PyTorch tensor, each image should be of shape (C, H, W), where C is a
tensor. In case of a NumPy array/PyTorch tensor, each image should be of shape (C, H, W), where C is a
number of channels, H and W are image height and width.
number of channels, H and W are image height and width.
return_tensors (`str` or [`~
file_
utils.TensorType`], *optional*, defaults to `'np'`):
return_tensors (`str` or [`~utils.TensorType`], *optional*, defaults to `'np'`):
If set, will return tensors of a particular framework. Acceptable values are:
If set, will return tensors of a particular framework. Acceptable values are:
- `'tf'`: Return TensorFlow `tf.constant` objects.
- `'tf'`: Return TensorFlow `tf.constant` objects.
...
...
src/transformers/models/vit/modeling_flax_vit.py
View file @
088c1880
...
@@ -79,7 +79,7 @@ VIT_INPUTS_DOCSTRING = r"""
...
@@ -79,7 +79,7 @@ VIT_INPUTS_DOCSTRING = r"""
Whether or not to return the hidden states of all layers. See `hidden_states` under returned tensors for
Whether or not to return the hidden states of all layers. See `hidden_states` under returned tensors for
more detail.
more detail.
return_dict (`bool`, *optional*):
return_dict (`bool`, *optional*):
Whether or not to return a [`~
file_
utils.ModelOutput`] instead of a plain tuple.
Whether or not to return a [`~utils.ModelOutput`] instead of a plain tuple.
"""
"""
...
...
src/transformers/models/vit/modeling_tf_vit.py
View file @
088c1880
...
@@ -626,8 +626,8 @@ VIT_INPUTS_DOCSTRING = r"""
...
@@ -626,8 +626,8 @@ VIT_INPUTS_DOCSTRING = r"""
interpolate_pos_encoding (`bool`, *optional*):
interpolate_pos_encoding (`bool`, *optional*):
Whether to interpolate the pre-trained position encodings.
Whether to interpolate the pre-trained position encodings.
return_dict (`bool`, *optional*):
return_dict (`bool`, *optional*):
Whether or not to return a [`~
file_
utils.ModelOutput`] instead of a plain tuple. This argument can be used
Whether or not to return a [`~utils.ModelOutput`] instead of a plain tuple. This argument can be used
in
in
eager mode, in graph mode the value will always be set to True.
eager mode, in graph mode the value will always be set to True.
training (`bool`, *optional*, defaults to `False``):
training (`bool`, *optional*, defaults to `False``):
Whether or not to use the model in training mode (some modules like dropout modules have different
Whether or not to use the model in training mode (some modules like dropout modules have different
behaviors between training and evaluation).
behaviors between training and evaluation).
...
...
src/transformers/models/vit/modeling_vit.py
View file @
088c1880
...
@@ -500,7 +500,7 @@ VIT_INPUTS_DOCSTRING = r"""
...
@@ -500,7 +500,7 @@ VIT_INPUTS_DOCSTRING = r"""
interpolate_pos_encoding (`bool`, *optional*):
interpolate_pos_encoding (`bool`, *optional*):
Whether to interpolate the pre-trained position encodings.
Whether to interpolate the pre-trained position encodings.
return_dict (`bool`, *optional*):
return_dict (`bool`, *optional*):
Whether or not to return a [`~
file_
utils.ModelOutput`] instead of a plain tuple.
Whether or not to return a [`~utils.ModelOutput`] instead of a plain tuple.
"""
"""
...
...
src/transformers/models/vit_mae/modeling_vit_mae.py
View file @
088c1880
...
@@ -631,7 +631,7 @@ VIT_MAE_INPUTS_DOCSTRING = r"""
...
@@ -631,7 +631,7 @@ VIT_MAE_INPUTS_DOCSTRING = r"""
Whether or not to return the hidden states of all layers. See `hidden_states` under returned tensors for
Whether or not to return the hidden states of all layers. See `hidden_states` under returned tensors for
more detail.
more detail.
return_dict (`bool`, *optional*):
return_dict (`bool`, *optional*):
Whether or not to return a [`~
file_
utils.ModelOutput`] instead of a plain tuple.
Whether or not to return a [`~utils.ModelOutput`] instead of a plain tuple.
"""
"""
...
...
src/transformers/models/wav2vec2/feature_extraction_wav2vec2.py
View file @
088c1880
...
@@ -118,7 +118,7 @@ class Wav2Vec2FeatureExtractor(SequenceFeatureExtractor):
...
@@ -118,7 +118,7 @@ class Wav2Vec2FeatureExtractor(SequenceFeatureExtractor):
raw_speech (`np.ndarray`, `List[float]`, `List[np.ndarray]`, `List[List[float]]`):
raw_speech (`np.ndarray`, `List[float]`, `List[np.ndarray]`, `List[List[float]]`):
The sequence or batch of sequences to be padded. Each sequence can be a numpy array, a list of float
The sequence or batch of sequences to be padded. Each sequence can be a numpy array, a list of float
values, a list of numpy arrays or a list of list of float values.
values, a list of numpy arrays or a list of list of float values.
padding (`bool`, `str` or [`~
file_
utils.PaddingStrategy`], *optional*, defaults to `False`):
padding (`bool`, `str` or [`~utils.PaddingStrategy`], *optional*, defaults to `False`):
Select a strategy to pad the returned sequences (according to the model's padding side and padding
Select a strategy to pad the returned sequences (according to the model's padding side and padding
index) among:
index) among:
...
@@ -156,7 +156,7 @@ class Wav2Vec2FeatureExtractor(SequenceFeatureExtractor):
...
@@ -156,7 +156,7 @@ class Wav2Vec2FeatureExtractor(SequenceFeatureExtractor):
</Tip>
</Tip>
return_tensors (`str` or [`~
file_
utils.TensorType`], *optional*):
return_tensors (`str` or [`~utils.TensorType`], *optional*):
If set, will return tensors instead of list of python integers. Acceptable values are:
If set, will return tensors instead of list of python integers. Acceptable values are:
- `'tf'`: Return TensorFlow `tf.constant` objects.
- `'tf'`: Return TensorFlow `tf.constant` objects.
...
...
src/transformers/models/wav2vec2/modeling_flax_wav2vec2.py
View file @
088c1880
...
@@ -281,7 +281,7 @@ WAV_2_VEC_2_INPUTS_DOCSTRING = r"""
...
@@ -281,7 +281,7 @@ WAV_2_VEC_2_INPUTS_DOCSTRING = r"""
Whether or not to return the hidden states of all layers. See `hidden_states` under returned tensors for
Whether or not to return the hidden states of all layers. See `hidden_states` under returned tensors for
more detail.
more detail.
return_dict (`bool`, *optional*):
return_dict (`bool`, *optional*):
Whether or not to return a [`~
file_
utils.ModelOutput`] instead of a plain tuple.
Whether or not to return a [`~utils.ModelOutput`] instead of a plain tuple.
"""
"""
...
...
src/transformers/models/wav2vec2/modeling_tf_wav2vec2.py
View file @
088c1880
...
@@ -1413,8 +1413,8 @@ WAV_2_VEC_2_INPUTS_DOCSTRING = r"""
...
@@ -1413,8 +1413,8 @@ WAV_2_VEC_2_INPUTS_DOCSTRING = r"""
more detail. This argument can be used only in eager mode, in graph mode the value in the config will be
more detail. This argument can be used only in eager mode, in graph mode the value in the config will be
used instead.
used instead.
return_dict (`bool`, *optional*):
return_dict (`bool`, *optional*):
Whether or not to return a [`~
file_
utils.ModelOutput`] instead of a plain tuple. This argument can be used
Whether or not to return a [`~utils.ModelOutput`] instead of a plain tuple. This argument can be used
in
in
eager mode, in graph mode the value will always be set to True.
eager mode, in graph mode the value will always be set to True.
training (`bool`, *optional*, defaults to `False``):
training (`bool`, *optional*, defaults to `False``):
Whether or not to use the model in training mode (some modules like dropout modules have different
Whether or not to use the model in training mode (some modules like dropout modules have different
behaviors between training and evaluation).
behaviors between training and evaluation).
...
...
src/transformers/models/wav2vec2/modeling_wav2vec2.py
View file @
088c1880
...
@@ -1226,7 +1226,7 @@ WAV_2_VEC_2_INPUTS_DOCSTRING = r"""
...
@@ -1226,7 +1226,7 @@ WAV_2_VEC_2_INPUTS_DOCSTRING = r"""
Whether or not to return the hidden states of all layers. See `hidden_states` under returned tensors for
Whether or not to return the hidden states of all layers. See `hidden_states` under returned tensors for
more detail.
more detail.
return_dict (`bool`, *optional*):
return_dict (`bool`, *optional*):
Whether or not to return a [`~
file_
utils.ModelOutput`] instead of a plain tuple.
Whether or not to return a [`~utils.ModelOutput`] instead of a plain tuple.
"""
"""
...
...
Prev
1
…
6
7
8
9
10
11
12
Next
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment