mirror of https://github.com/coqui-ai/TTS.git
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.pypull/1776/head
parent
4f31402227
commit
903d9c791a
|
@ -514,7 +514,7 @@ class GlowTTS(BaseTTS):
|
|||
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):
|
||||
|
|
Loading…
Reference in New Issue