refine decoding process, merge flow and vocoder

This commit is contained in:
志浩 2024-07-10 16:36:01 +08:00
parent 04c2d487ed
commit 4f2ca562e9

View File

@ -464,10 +464,7 @@ class HifiGan(nn.Module):
"""
recon_speech = self._decode(x.transpose(1, 2)).squeeze(1)
retval = dict(
recon_speech=recon_speech,
)
return retval
return recon_speech
def collect_feats(self, **batch: torch.Tensor) -> Dict[str, torch.Tensor]:
pass