denig

2023-05-18 02:16:40 +08:00 · 2023-05-18 02:16:40 +08:00 · 49e933e8ae
parent d37bddd455
commit 49e933e8ae
1 changed files with 3 additions and 1 deletions
--- a/inference/infer_tool.py
+++ b/inference/infer_tool.py
@ -249,6 +249,7 @@ class Svc(object):
                    spk_mix_dict = None,
                    gt_spec=audio_mel,
                    infer=True, 
+                    aug_shift=0,
                    infer_speedup=self.diffusion_args.infer.speedup, 
                    method=self.diffusion_args.infer.methold,
                    k_step=k_step)
@ -265,7 +266,8 @@ class Svc(object):
                    spk_id = sid, 
                    spk_mix_dict = None,
                    gt_spec=None,
-                    infer=True, 
+                    infer=True,
+                    aug_shift=0, 
                    infer_speedup=self.diffusion_args.infer.speedup, 
                    method=self.diffusion_args.infer.methold,
                    k_step=k_step)