wassname/TTS
1
0
Fork 0
mirror of https://github.com/wassname/TTS.git synced 2026-09-11 12:00:24 +08:00
Code Issues Packages Projects Releases Wiki Activity

update sequence_mask import globally

Browse Source
This commit is contained in:
Eren Gölge
2021-06-28 17:03:19 +02:00
parent b2218e882a
commit 9134c7dfb6
8 changed files with 11 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/tts_tests/test_feed_forward_layers.py
+1 -1
View File
@@ -2,7 +2,7 @@ import torch
from TTS.tts.layers.feed_forward.decoder import Decoder
from TTS.tts.layers.feed_forward.encoder import Encoder
from TTS.tts.utils.generic_utils import sequence_mask
from TTS.tts.utils.data import sequence_mask
device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
tests/tts_tests/test_speedy_speech_layers.py
+1 -1
View File
@@ -2,7 +2,7 @@ import torch
from TTS.tts.layers.feed_forward.duration_predictor import DurationPredictor
from TTS.tts.models.speedy_speech import SpeedySpeech
from TTS.tts.utils.generic_utils import sequence_mask
from TTS.tts.utils.data import sequence_mask
use_cuda = torch.cuda.is_available()
device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
tests/tts_tests/test_tacotron_layers.py
+1 -1
View File
@@ -4,7 +4,7 @@ import torch as T
from TTS.tts.layers.losses import L1LossMasked, SSIMLoss
from TTS.tts.layers.tacotron.tacotron import CBHG, Decoder, Encoder, Prenet
from TTS.tts.utils.generic_utils import sequence_mask
from TTS.tts.utils.data import sequence_mask
# pylint: disable=unused-variable