opensource/ChatTTS
3
0
Fork 0
mirror of https://github.com/2noise/ChatTTS.git synced 2026-08-28 17:45:02 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat(test): use manual_seed param

Browse Source
This commit is contained in:
源文雨
2024-08-03 17:15:53 +08:00
parent c140a0b691
commit 47b476e198
1 changed files with 8 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/#655.py
+8 -9
View File
@@ -14,7 +14,6 @@ import ChatTTS
from tools.logger import get_logger
from tools.normalizer import normalizer_en_nemo_text
from tools.seeder import TorchSeedContext
logger = get_logger("Test", lv=logging.WARN)
@@ -32,14 +31,14 @@ text = ["What is [uv_break]your favorite english food?[laugh][lbreak]"]
fail = False
with TorchSeedContext(12345):
refined_text = chat.infer(
text,
refine_text_only=True,
params_refine_text=ChatTTS.Chat.RefineTextParams(
prompt="[oral_2][laugh_0][break_6]",
),
)
refined_text = chat.infer(
text,
refine_text_only=True,
params_refine_text=ChatTTS.Chat.RefineTextParams(
prompt="[oral_2][laugh_0][break_6]",
manual_seed=12345,
),
)
if (
refined_text[0]
!= "what is [uv_break] your favorite [uv_break] english food [laugh] like [laugh]"