Medix-AI/coqui-tts
Medix-AI
/
coqui-tts
mirror of https://github.com/coqui-ai/TTS.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Fix appending a new block to input_blocks

This commit is contained in:
Subuday 2024-02-15 08:55:52 +00:00
parent fd6c0afbbf
commit 8676ab30d9
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
TTS/tts/layers/matcha_tts/UNet.py
View File

@ -76,7 +76,7 @@ class UNet(nn.Module):
) )
self.input_blocks = nn.ModuleList([]) self.input_blocks = nn.ModuleList([])
block_in_channels = in_channels block_in_channels = in_channels * 2
for _ in range(num_blocks): for _ in range(num_blocks):
block = nn.ModuleList([]) block = nn.ModuleList([])
@ -88,6 +88,8 @@ class UNet(nn.Module):
) )
) )
self.input_blocks.append(block)
self.middle_blocks = nn.ModuleList([]) self.middle_blocks = nn.ModuleList([])
self.output_blocks = nn.ModuleList([]) self.output_blocks = nn.ModuleList([])