From ee7d6694ed1d6cf861ed21266e4e82e449ebbcba Mon Sep 17 00:00:00 2001
From: Yoach Lacombe <52246514+ylacombe@users.noreply.github.com>
Date: Mon, 14 Aug 2023 16:41:47 +0200
Subject: [PATCH] Set can_generate for SpeechT5ForTextToSpeech (#25493)

add can_generate=True to SpeechT5ForTextToSpeech
---
 src/transformers/models/speecht5/modeling_speecht5.py | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/src/transformers/models/speecht5/modeling_speecht5.py b/src/transformers/models/speecht5/modeling_speecht5.py
index f4e0e8052c3..8471cb76ae7 100644
--- 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,