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
f65307e4
Unverified
Commit
f65307e4
authored
Jul 26, 2022
by
Yih-Dar
Committed by
GitHub
Jul 26, 2022
Browse files
Fix dtype of input_features in docstring (#18258)
Co-authored-by:
ydshieh
<
ydshieh@users.noreply.github.com
>
parent
bd87480d
Changes
2
Show whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
2 additions
and
2 deletions
+2
-2
src/transformers/models/speech_encoder_decoder/modeling_speech_encoder_decoder.py
...speech_encoder_decoder/modeling_speech_encoder_decoder.py
+1
-1
src/transformers/models/speech_to_text/modeling_speech_to_text.py
...sformers/models/speech_to_text/modeling_speech_to_text.py
+1
-1
No files found.
src/transformers/models/speech_encoder_decoder/modeling_speech_encoder_decoder.py
View file @
f65307e4
...
@@ -135,7 +135,7 @@ SPEECH_ENCODER_DECODER_INPUTS_DOCSTRING = r"""
...
@@ -135,7 +135,7 @@ SPEECH_ENCODER_DECODER_INPUTS_DOCSTRING = r"""
into an array of type *List[float]* or a *numpy.ndarray*, *e.g.* via the soundfile library (*pip install
into an array of type *List[float]* or a *numpy.ndarray*, *e.g.* via the soundfile library (*pip install
soundfile*). To prepare the array into *input_values*, the [`Wav2Vec2Processor`] should be used for padding
soundfile*). To prepare the array into *input_values*, the [`Wav2Vec2Processor`] should be used for padding
and conversion into a tensor of type *torch.FloatTensor*. See [`Wav2Vec2Processor.__call__`] for details.
and conversion into a tensor of type *torch.FloatTensor*. See [`Wav2Vec2Processor.__call__`] for details.
input_features (`torch.
Long
Tensor` of shape `(batch_size, sequence_length, feature_size)`, *optional*):
input_features (`torch.
Float
Tensor` of shape `(batch_size, sequence_length, feature_size)`, *optional*):
Float values of fbank features extracted from the raw speech waveform. Raw speech waveform can be obtained
Float values of fbank features extracted from the raw speech waveform. Raw speech waveform can be obtained
by loading a `.flac` or `.wav` audio file into an array of type `List[float]` or a `numpy.ndarray`, *e.g.*
by loading a `.flac` or `.wav` audio file into an array of type `List[float]` or a `numpy.ndarray`, *e.g.*
via the soundfile library (`pip install soundfile`). To prepare the array into `input_features`, the
via the soundfile library (`pip install soundfile`). To prepare the array into `input_features`, the
...
...
src/transformers/models/speech_to_text/modeling_speech_to_text.py
View file @
f65307e4
...
@@ -599,7 +599,7 @@ SPEECH_TO_TEXT_START_DOCSTRING = r"""
...
@@ -599,7 +599,7 @@ SPEECH_TO_TEXT_START_DOCSTRING = r"""
SPEECH_TO_TEXT_INPUTS_DOCSTRING
=
r
"""
SPEECH_TO_TEXT_INPUTS_DOCSTRING
=
r
"""
Args:
Args:
input_features (`torch.
Long
Tensor` of shape `(batch_size, sequence_length, feature_size)`):
input_features (`torch.
Float
Tensor` of shape `(batch_size, sequence_length, feature_size)`):
Float values of fbank features extracted from the raw speech waveform. Raw speech waveform can be obtained
Float values of fbank features extracted from the raw speech waveform. Raw speech waveform can be obtained
by loading a `.flac` or `.wav` audio file into an array of type `List[float]` or a `numpy.ndarray`, *e.g.*
by loading a `.flac` or `.wav` audio file into an array of type `List[float]` or a `numpy.ndarray`, *e.g.*
via the soundfile library (`pip install soundfile`). To prepare the array into `input_features`, the
via the soundfile library (`pip install soundfile`). To prepare the array into `input_features`, the
...
...
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