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

Update test_onnx.py

Browse Source
This commit is contained in:
jinzr 2023-12-05 14:40:54 +08:00
parent eb686b8da3
commit fddfd2466f
1 changed files with 5 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
egs/vctk/TTS/vits/test_onnx.py
View File

@ -79,7 +79,9 @@ class OnnxModel:
) )
logging.info(f"{self.model.get_modelmeta().custom_metadata_map}") logging.info(f"{self.model.get_modelmeta().custom_metadata_map}")
def __call__(self, tokens: torch.Tensor, tokens_lens: torch.Tensor) -> torch.Tensor: def __call__(
self, tokens: torch.Tensor, tokens_lens: torch.Tensor, speaker: torch.Tensor
) -> torch.Tensor:
""" """
Args: Args:
tokens: tokens:
@ -100,7 +102,8 @@ class OnnxModel:
self.model.get_inputs()[1].name: tokens_lens.numpy(), self.model.get_inputs()[1].name: tokens_lens.numpy(),
self.model.get_inputs()[2].name: noise_scale.numpy(), self.model.get_inputs()[2].name: noise_scale.numpy(),
self.model.get_inputs()[3].name: noise_scale_dur.numpy(), self.model.get_inputs()[3].name: noise_scale_dur.numpy(),
self.model.get_inputs()[4].name: alpha.numpy(), self.model.get_inputs()[4].name: speaker.numpy(),
self.model.get_inputs()[5].name: alpha.numpy(),
}, },
)[0] )[0]
return torch.from_numpy(out) return torch.from_numpy(out)