Skip to content

Commit ceef28c

Browse files
committed
add todo note for a future fix
1 parent 1d2667f commit ceef28c

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

Modules/ToucanTTS/dit.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -126,7 +126,7 @@ def forward(self, x, c, x_mask):
126126
else:
127127
# no condition version
128128
x = x + self.attn(self.norm1(x.transpose(1, 2)).transpose(1, 2), attn_mask)
129-
x = x + self.mlp(self.norm1(x.transpose(1, 2)).transpose(1, 2), x_mask)
129+
x = x + self.mlp(self.norm1(x.transpose(1, 2)).transpose(1, 2), x_mask) # TODO this is technically a bug, but it seems to work ok with the bug. Before the next major model is trained for a release, norm1 should be changed to norm2 in this line.
130130
return x
131131

132132
@staticmethod

0 commit comments

Comments
 (0)