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")
|
raise ImportError("Could not install required packages")
|
||||||
|
|
||||||
try:
|
try:
|
||||||
from f5_tts.model import CFM, DiT
|
from f5_tts.model import CFM, DiT, UNetT
|
||||||
from f5_tts.model.utils import (
|
from f5_tts.model.utils import (
|
||||||
get_tokenizer,
|
get_tokenizer,
|
||||||
convert_char_to_pinyin,
|
convert_char_to_pinyin,
|
||||||
@@ -1102,4 +1102,4 @@ class tts_class:
|
|||||||
except Exception as e:
|
except Exception as e:
|
||||||
print(f"[{self.branding}Debug] Device verification warning: {str(e)}") if self.debug_tts else None
|
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