Skip to content

Commit 07a5d8a

Browse files
authored
Fix compilation error on Colab T4 GPU.
1 parent 53d0927 commit 07a5d8a

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

vietTTS/nat/config.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -15,7 +15,8 @@ class FLAGS(Namespace):
1515
acoustic_encoder_dim = 256
1616

1717
# dataset
18-
max_phoneme_seq_len = 128 * 3
18+
max_phoneme_seq_len = 256 * 1
19+
assert max_phoneme_seq_len % 256 == 0 # prevent compilation error on Colab T4 GPU
1920
max_wave_len = 1024 * 64 * 3
2021

2122
# Montreal Forced Aligner

0 commit comments

Comments
 (0)