Skip to content

Commit

Permalink
Fix SpeechT5 forward docstrings (#30287)
Browse files Browse the repository at this point in the history
  • Loading branch information
ylacombe authored and ArthurZucker committed Apr 22, 2024
1 parent d899edd commit 02419d0
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion src/transformers/models/speecht5/modeling_speecht5.py
Original file line number Diff line number Diff line change
Expand Up @@ -2687,7 +2687,7 @@ def forward(
>>> set_seed(555) # make deterministic
>>> # 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
torch.Size([15872])
```
Expand Down

0 comments on commit 02419d0

Please sign in to comment.