From 4679bd39765b403dfc573f4d26a23d943494de0a Mon Sep 17 00:00:00 2001
From: Andres Marafioti <andimarafioti@gmail.com>
Date: Thu, 22 Aug 2024 13:57:55 +0200
Subject: [PATCH] only import melo if needed
---
s2s_pipeline.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/s2s_pipeline.py b/s2s_pipeline.py
index 66aeaa5..b08d0f9 100644
--- a/s2s_pipeline.py
+++ b/s2s_pipeline.py
@@ -12,7 +12,6 @@ from time import perf_counter
from typing import Optional
from sys import platform
from LLM.mlx_lm import MLXLanguageModelHandler
-from TTS.melotts import MeloTTSHandler
from baseHandler import BaseHandler
from STT.lightning_whisper_mlx_handler import LightningWhisperSTTHandler
import numpy as np
@@ -1069,6 +1068,7 @@ def main():
)
elif module_kwargs.tts == 'melo':
+ from TTS.melotts import MeloTTSHandler
tts = MeloTTSHandler(
stop_event,
queue_in=lm_response_queue,
--
GitLab