Skip to content

GitLab

"vscode:/vscode.git/clone" did not exist on "464aa74659b9711d2d64159ab82e1c49fa739fb7"
Unverified Commit 41145247 authored by Yoach Lacombe's avatar Yoach Lacombe Committed by GitHub
Browse files

Fix SpeechT5 forward docstrings (#30287)

parent 40eb6d6c
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
src/transformers/models/speecht5/modeling_speecht5.py
View file @ 41145247
...@@ -2687,7 +2687,7 @@ class SpeechT5ForTextToSpeech(SpeechT5PreTrainedModel): ...@@ -2687,7 +2687,7 @@ class SpeechT5ForTextToSpeech(SpeechT5PreTrainedModel):
>>> set_seed(555) # make deterministic >>> set_seed(555) # make deterministic
>>> # generate speech >>> # generate speech
>>> speech = model.generate(inputs["input_ids"], speaker_embeddings, vocoder=vocoder) >>> speech = model.generate(inputs["input_ids"], speaker_embeddings=speaker_embeddings, vocoder=vocoder)
>>> speech.shape >>> speech.shape
torch.Size([15872]) torch.Size([15872])
``` ```
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment