wassname/TTS
1
0
Fork 0
mirror of https://github.com/wassname/TTS.git synced 2026-09-09 11:16:00 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fix VITS stochastic duration predictor

Browse Source
This commit is contained in:
Eren Gölge
2021-11-08 09:20:11 +01:00
parent 3a77899775
commit b6b14a76af
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
TTS/tts/layers/vits/stochastic_duration_predictor.py
+1 -1
View File
@@ -266,7 +266,7 @@ class StochasticDurationPredictor(nn.Module):
flows = list(reversed(self.flows))
flows = flows[:-2] + [flows[-1]] # remove a useless vflow
z = torch.rand(x.size(0), 2, x.size(2)).to(device=x.device, dtype=x.dtype) * noise_scale
z = torch.randn(x.size(0), 2, x.size(2)).to(device=x.device, dtype=x.dtype) * noise_scale
for flow in flows:
z = torch.flip(z, [1])
z = flow(z, x_mask, g=x, reverse=reverse)