Skip to content

Commit 709639f

Browse files
committed
Fix test for nyanko
1 parent 0ac3342 commit 709639f

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tests/test_nyanko.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,7 @@
1616
from deepvoice3_pytorch.builder import nyanko
1717
from deepvoice3_pytorch import MultiSpeakerTTSModel, AttentionSeq2Seq
1818

19-
use_cuda = torch.cuda.is_available()
19+
use_cuda = torch.cuda.is_available() and False
2020
num_mels = 80
2121
num_freq = 513
2222
outputs_per_step = 4
@@ -95,7 +95,7 @@ def test_incremental_correctness():
9595
text_positions = np.arange(1, len(seqs[0]) + 1).reshape(1, len(seqs[0]))
9696

9797
mel_path = join(dirname(__file__), "data", "ljspeech-mel-00001.npy")
98-
mel = np.load(mel_path)
98+
mel = np.load(mel_path)[::4]
9999
max_target_len = mel.shape[0]
100100
r = 1
101101
mel_dim = 80

0 commit comments

Comments
 (0)