riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-07-30 17:52:35 +06:00
Code Issues Actions 30 Packages Projects Releases Wiki Activity

fix SpeechT5 doc comments (#22854)

Browse Source 
fix doc comments
This commit is contained in:
Matthijs Hollemans 2023-04-19 14:10:40 +02:00 committed by GitHub
parent 84a6570e7b
commit c582e8aad0
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -2450,7 +2450,7 @@ class SpeechT5ForSpeechToText(SpeechT5PreTrainedModel):
>>> # compute loss
>>> loss = model(**inputs).loss
>>> round(loss.item(), 2)
19.88
19.68
```
"""
return_dict = return_dict if return_dict is not None else self.config.use_return_dict
@ -2724,7 +2724,7 @@ class SpeechT5ForTextToSpeech(SpeechT5PreTrainedModel):
>>> # generate speech
>>> speech = model.generate_speech(inputs["input_ids"], speaker_embeddings, vocoder=vocoder)
>>> speech.shape
torch.Size([16384])
torch.Size([15872])
```
"""
return_dict = return_dict if return_dict is not None else self.config.use_return_dict