kromskii2/alltalk_tts
1
0 Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

add UNetT import

Browse Source
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
+1 -1
Download Patch File Download Diff File Expand all files Collapse all files
system/tts_engines/f5tts/model_engine.py
+1 -1
View File
@@ -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,