Spaces:

utkarsh-dixit
/

WhisperFusion

Paused

jpc commited on Jan 24

Commit

f97a5dd

•

1 Parent(s): 7bf3a05

Enable torch.compile for TTS

Files changed (1) hide show

tts_service.py CHANGED Viewed

@@ -11,7 +11,7 @@ class WhisperSpeechTTS:
         pass
     def initialize_model(self):
-        self.pipe = Pipeline(s2a_ref='collabora/whisperspeech:s2a-q4-tiny-en+pl.model')
         self.last_llm_response = None
     def run(self, host, port, audio_queue=None):

         pass
     def initialize_model(self):
+        self.pipe = Pipeline(s2a_ref='collabora/whisperspeech:s2a-q4-tiny-en+pl.model', torch_compile=True)
         self.last_llm_response = None
     def run(self, host, port, audio_queue=None):