mirrors/icefall
1
0
Fork 0
mirror of https://github.com/k2-fsa/icefall.git synced 2025-08-08 17:42:21 +00:00
Code Issues Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
zr_jin 2023-12-08 21:11:31 +08:00 committed by GitHub
parent e9ec827de7
commit df56aff31e
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
egs/ljspeech/TTS/vits/export-onnx.py
View File

@ -115,8 +115,8 @@ class OnnxModel(nn.Module):
tokens: torch.Tensor, tokens: torch.Tensor,
tokens_lens: torch.Tensor, tokens_lens: torch.Tensor,
noise_scale: float = 0.667, noise_scale: float = 0.667,
noise_scale_dur: float = 0.8,
alpha: float = 1.0, alpha: float = 1.0,
noise_scale_dur: float = 0.8,
) -> Tuple[torch.Tensor, torch.Tensor]: ) -> Tuple[torch.Tensor, torch.Tensor]:
"""Please see the help information of VITS.inference_batch """Please see the help information of VITS.inference_batch

2
egs/vctk/TTS/vits/export-onnx.py
View File

@ -121,9 +121,9 @@ class OnnxModel(nn.Module):
tokens: torch.Tensor, tokens: torch.Tensor,
tokens_lens: torch.Tensor, tokens_lens: torch.Tensor,
noise_scale: float = 0.667, noise_scale: float = 0.667,
alpha: float = 1.0,
noise_scale_dur: float = 0.8, noise_scale_dur: float = 0.8,
speaker: int = 20, speaker: int = 20,
alpha: float = 1.0,
) -> Tuple[torch.Tensor, torch.Tensor]: ) -> Tuple[torch.Tensor, torch.Tensor]:
"""Please see the help information of VITS.inference_batch """Please see the help information of VITS.inference_batch