From b7758f25a940960ce64e4b656e32c5a5bb44a754 Mon Sep 17 00:00:00 2001 From: James Betker Date: Thu, 16 Jun 2022 20:06:27 -0600 Subject: [PATCH] and fix --- codes/models/audio/music/gpt_music2.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/codes/models/audio/music/gpt_music2.py b/codes/models/audio/music/gpt_music2.py index 3a773646..dc06dcbe 100644 --- a/codes/models/audio/music/gpt_music2.py +++ b/codes/models/audio/music/gpt_music2.py @@ -24,8 +24,8 @@ class UpperEncoder(nn.Module): ResBlock(spec_dim, out_channels=edim(1), use_conv=True, dims=1, down=True), ResBlock(edim(1), out_channels=edim(2), use_conv=True, dims=1, down=True), ResBlock(edim(2), out_channels=edim(3), use_conv=True, dims=1, down=True), - ResBlock(edim(5), out_channels=edim(5), use_conv=True, dims=1), - ResBlock(edim(5), out_channels=hidden_dim, use_conv=True, dims=1, down=True)) + ResBlock(edim(3), out_channels=edim(4), use_conv=True, dims=1), + ResBlock(edim(4), out_channels=hidden_dim, use_conv=True, dims=1, down=True)) self.encoder = nn.Sequential( AttentionBlock(hidden_dim, 4, do_activation=True), ResBlock(hidden_dim, out_channels=hidden_dim, use_conv=True, dims=1),