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

update regarding torch 1.2

This commit is contained in:
Eren Golge 2019-08-13 12:14:34 +02:00
parent 3cbbd8d6e0
commit 64f2b95c31
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
layers/common_layers.py
View File

@ -240,7 +240,7 @@ class Attention(nn.Module):
attention_hidden_state, processed_inputs) attention_hidden_state, processed_inputs)
# apply masking # apply masking
if mask is not None: if mask is not None:
attention.data.masked_fill_(1 - mask, self._mask_value) attention.data.masked_fill_(torch.bitwise_not(mask), self._mask_value)
# apply windowing - only in eval mode # apply windowing - only in eval mode
if not self.training and self.windowing: if not self.training and self.windowing:
attention = self.apply_windowing(attention, inputs) attention = self.apply_windowing(attention, inputs)