From 590eb23e782b36f02ebd0798cfd29148d8d4cc6a Mon Sep 17 00:00:00 2001
From: erew123 <35898566+erew123@users.noreply.github.com>
Date: Tue, 5 Nov 2024 10:25:32 +0000
Subject: [PATCH] add UNetT import

---
 system/tts_engines/f5tts/model_engine.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/system/tts_engines/f5tts/model_engine.py b/system/tts_engines/f5tts/model_engine.py
index 896af53..88dab71 100644
--- a/system/tts_engines/f5tts/model_engine.py
+++ b/system/tts_engines/f5tts/model_engine.py
@@ -79,7 +79,7 @@ def install_and_restart():
         raise ImportError("Could not install required packages")
 
 try:
-    from f5_tts.model import CFM, DiT
+    from f5_tts.model import CFM, DiT, UNetT
     from f5_tts.model.utils import (
         get_tokenizer,
         convert_char_to_pinyin,
@@ -1102,4 +1102,4 @@ class tts_class:
             except Exception as e:
                 print(f"[{self.branding}Debug] Device verification warning: {str(e)}") if self.debug_tts else None
             
-            self.tts_generating_lock = False
\ No newline at end of file
+            self.tts_generating_lock = False