riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-08-01 02:31:11 +06:00
Code Issues Actions 3 Packages Projects Releases Wiki Activity

Fix SpeechT5 forward docstrings (#30287)

Browse Source
This commit is contained in:
Yoach Lacombe 2024-04-17 11:23:49 +02:00 committed by GitHub
parent 40eb6d6c5f
commit 4114524706
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/transformers/models/speecht5/modeling_speecht5.py
View File

@ -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])
``` ```