Fix for FloorDiv Function Warning (#1760)

* Fix for Floor Function Warning Fix for Floor Function Warning * Adding double quotes to fix formatting Adding double quotes to fix formatting * Update glow_tts.py * Update glow_tts.py
coqui-ai · Jul 20, 2022 · 903d9c7 · 903d9c7
1 parent 4f31402
commit 903d9c7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/TTS/tts/models/glow_tts.py b/TTS/tts/models/glow_tts.py
@@ -514,7 +514,7 @@ def preprocess(self, y, y_lengths, y_max_length, attn=None):
             y = y[:, :, :y_max_length]
             if attn is not None:
                 attn = attn[:, :, :, :y_max_length]
-        y_lengths = (y_lengths // self.num_squeeze) * self.num_squeeze
+        y_lengths = torch.div(y_lengths, self.num_squeeze, rounding_mode="floor") * self.num_squeeze
         return y, y_lengths, y_max_length, attn
 
     def store_inverse(self):