Set can_generate for SpeechT5ForTextToSpeech (#25493)

add can_generate=True to SpeechT5ForTextToSpeech

Set can_generate for SpeechT5ForTextToSpeech (#25493)
add can_generate=True to SpeechT5ForTextToSpeech
ee7d6694 · Yoach Lacombe · GitHub · 87c9d8a1 · ee7d6694
Unverified Commit ee7d6694 authored Aug 14, 2023 by Yoach Lacombe Committed by GitHub Aug 14, 2023
Show whitespace changes
Inline Side-by-side

Showing with 7 additions and 0 deletions

src/transformers/models/speecht5/modeling_speecht5.py src/transformers/models/speecht5/modeling_speecht5.py +7 -0

No files found.
--- a/src/transformers/models/speecht5/modeling_speecht5.py
+++ b/src/transformers/models/speecht5/modeling_speecht5.py
@@ -2783,6 +2783,13 @@ class SpeechT5ForTextToSpeech(SpeechT5PreTrainedModel):
            encoder_attentions=outputs.encoder_attentions,
        )

+    def can_generate(self) -> bool:
+        """
+        Returns True. This model can `generate` and must therefore have this property set to True in order to be used
+        in the TTS pipeline.
+        """
+        return True
+
    @torch.no_grad()
    def generate(
        self,