Skip to content

GitLab

Unverified Commit 41145247 authored by Yoach Lacombe's avatar Yoach Lacombe Committed by GitHub
Browse files

Fix SpeechT5 forward docstrings (#30287)

parent 40eb6d6c
Hide 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