Change the VITS upsampling interpolation trick to linear

coqui-ai · May 9, 2022 · f8bafdf · f8bafdf
1 parent 9014a99
commit f8bafdf
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/TTS/tts/models/vits.py b/TTS/tts/models/vits.py
@@ -890,9 +890,7 @@ def upsampling_z(self, z, slice_ids=None, y_lengths=None, y_mask=None):
             spec_segment_size = spec_segment_size * int(self.interpolate_factor)
             # interpolate z if needed
             if self.args.interpolate_z:
-                z = torch.nn.functional.interpolate(
-                    z.unsqueeze(0), scale_factor=[1, self.interpolate_factor], mode="nearest"
-                ).squeeze(0)
+                z = torch.nn.functional.interpolate(z, scale_factor=[self.interpolate_factor], mode="linear").squeeze(0)
                 # recompute the mask if needed
                 if y_lengths is not None and y_mask is not None:
                     y_mask = (