From c582e8aad0a7eaeb3430efd0bcb53e9ed66a2f72 Mon Sep 17 00:00:00 2001 From: Matthijs Hollemans Date: Wed, 19 Apr 2023 14:10:40 +0200 Subject: [PATCH] fix SpeechT5 doc comments (#22854) fix doc comments --- src/transformers/models/speecht5/modeling_speecht5.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/transformers/models/speecht5/modeling_speecht5.py b/src/transformers/models/speecht5/modeling_speecht5.py index 819d8948cab..786110c89da 100644 --- a/src/transformers/models/speecht5/modeling_speecht5.py +++ b/src/transformers/models/speecht5/modeling_speecht5.py @@ -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