add UNetT import

This commit is contained in:
erew123 authored and GitHub committed 2024-11-05 10:25:32 +00:00
1 parent 1b4e982955
commit 590eb23e78
1 file changed
+2 -2
+2 -2
View File
@@ -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
self.tts_generating_lock = False