Medix-AI/coqui-tts
Medix-AI
/
coqui-tts
mirror of https://github.com/coqui-ai/TTS.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Change the VITS upsampling interpolation trick to linear (#1564)

This commit is contained in:
Edresson Casanova 2022-05-13 05:52:39 -03:00 committed by GitHub
parent c6008e5235
commit e5d8ec2402
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
TTS/tts/models/vits.py
View File

@ -890,9 +890,7 @@ class Vits(BaseTTS):
spec_segment_size = spec_segment_size * int(self.interpolate_factor) spec_segment_size = spec_segment_size * int(self.interpolate_factor)
# interpolate z if needed # interpolate z if needed
if self.args.interpolate_z: if self.args.interpolate_z:
z = torch.nn.functional.interpolate( z = torch.nn.functional.interpolate(z, scale_factor=[self.interpolate_factor], mode="linear").squeeze(0)
z.unsqueeze(0), scale_factor=[1, self.interpolate_factor], mode="nearest"
).squeeze(0)
# recompute the mask if needed # recompute the mask if needed
if y_lengths is not None and y_mask is not None: if y_lengths is not None and y_mask is not None:
y_mask = ( y_mask = (