minor fixes to the vits onnx exportation scripts (#1408)

k2-fsa · Dec 8, 2023 · df56aff · df56aff
1 parent e9ec827
commit df56aff
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/egs/ljspeech/TTS/vits/export-onnx.py b/egs/ljspeech/TTS/vits/export-onnx.py
@@ -115,8 +115,8 @@ def forward(
         tokens: torch.Tensor,
         tokens_lens: torch.Tensor,
         noise_scale: float = 0.667,
-        noise_scale_dur: float = 0.8,
         alpha: float = 1.0,
+        noise_scale_dur: float = 0.8,
     ) -> Tuple[torch.Tensor, torch.Tensor]:
         """Please see the help information of VITS.inference_batch
 

diff --git a/egs/vctk/TTS/vits/export-onnx.py b/egs/vctk/TTS/vits/export-onnx.py
@@ -121,9 +121,9 @@ def forward(
         tokens: torch.Tensor,
         tokens_lens: torch.Tensor,
         noise_scale: float = 0.667,
+        alpha: float = 1.0,
         noise_scale_dur: float = 0.8,
         speaker: int = 20,
-        alpha: float = 1.0,
     ) -> Tuple[torch.Tensor, torch.Tensor]:
         """Please see the help information of VITS.inference_batch