feat(test): use manual_seed param

2noise · Aug 3, 2024 · 47b476e · 47b476e
1 parent c140a0b
commit 47b476e
Showing 1 changed file with 8 additions and 9 deletions.
diff --git a/tests/#655.py b/tests/#655.py
@@ -14,7 +14,6 @@
 
 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 @@
 
 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]"