mirrors/icefall
1
0
Fork 0
mirror of https://github.com/k2-fsa/icefall.git synced 2025-09-04 14:44:18 +00:00
Code Issues Packages Projects Releases Wiki Activity

minor fixes

Browse Source
This commit is contained in:
jinzr 2023-12-08 20:57:51 +08:00
parent e9ec827de7
commit e748d95a70
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_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

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

@ -121,9 +121,9 @@ class OnnxModel(nn.Module):
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