opensource/ChatTTS
3
0
Fork 0
mirror of https://github.com/2noise/ChatTTS.git synced 2026-08-30 17:05:27 +08:00
Code Issues Packages Projects Releases Wiki Activity

refactor: remove unnecessary transpositions (#488)

Browse Source
This commit is contained in:
charSLee013
2024-06-28 18:01:20 +08:00
committed by GitHub
parent bda9b5a607
commit e04ec2fcce
1 changed files with 5 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files
ChatTTS/model/dvae.py
+5 -6
View File
@@ -154,9 +154,8 @@ class DVAEDecoder(nn.Module):
)
self.conv_out = nn.Conv1d(hidden, odim, kernel_size=1, bias=False)
def forward(self, input: torch.Tensor, conditioning=None) -> torch.Tensor:
# B, T, C
x = input.transpose_(1, 2)
def forward(self, x: torch.Tensor, conditioning=None) -> torch.Tensor:
# B, C, T
y = self.conv_in(x)
del x
for f in self.decoder_block:
@@ -164,7 +163,7 @@ class DVAEDecoder(nn.Module):
x = self.conv_out(y)
del y
return x.transpose_(1, 2)
return x
class DVAE(nn.Module):
@@ -214,8 +213,8 @@ class DVAE(nn.Module):
dec_out = self.out_conv(
self.decoder(
input=vq_feats.transpose_(1, 2),
).transpose_(1, 2),
x=vq_feats,
),
)
return torch.mul(dec_out, self.coef, out=dec_out)