From 8676ab30d9b6962621a1f1ba9b5ca886723f5bca Mon Sep 17 00:00:00 2001 From: Subuday Date: Thu, 15 Feb 2024 08:55:52 +0000 Subject: [PATCH] Fix appending a new block to input_blocks --- TTS/tts/layers/matcha_tts/UNet.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/TTS/tts/layers/matcha_tts/UNet.py b/TTS/tts/layers/matcha_tts/UNet.py index 8547bb9b..0183c787 100644 --- a/TTS/tts/layers/matcha_tts/UNet.py +++ b/TTS/tts/layers/matcha_tts/UNet.py @@ -76,7 +76,7 @@ class UNet(nn.Module): ) self.input_blocks = nn.ModuleList([]) - block_in_channels = in_channels + block_in_channels = in_channels * 2 for _ in range(num_blocks): block = nn.ModuleList([]) @@ -88,6 +88,8 @@ class UNet(nn.Module): ) ) + self.input_blocks.append(block) + self.middle_blocks = nn.ModuleList([]) self.output_blocks = nn.ModuleList([])