add UNetT import
This commit is contained in:
1 parent
1b4e982955
commit
590eb23e78
1 file changed
+2
-2
@@ -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
|
||||
Reference in new issue
Block a user