From b13d983c24277565e76ce62c8953de04fb584369 Mon Sep 17 00:00:00 2001 From: James Betker Date: Fri, 6 May 2022 00:25:27 -0600 Subject: [PATCH] and mel_head --- codes/models/audio/music/music_gen_fill_gaps.py | 1 - 1 file changed, 1 deletion(-) diff --git a/codes/models/audio/music/music_gen_fill_gaps.py b/codes/models/audio/music/music_gen_fill_gaps.py index 45dd0624..b2761763 100644 --- a/codes/models/audio/music/music_gen_fill_gaps.py +++ b/codes/models/audio/music/music_gen_fill_gaps.py @@ -151,7 +151,6 @@ class MusicGenerator(nn.Module): DiffusionLayer(model_channels, dropout, num_heads), ) self.integrating_conv = nn.Conv1d(model_channels*2, model_channels, kernel_size=1) - self.mel_head = nn.Conv1d(model_channels, in_channels, kernel_size=3, padding=1) self.layers = nn.ModuleList([DiffusionLayer(model_channels, dropout, num_heads) for _ in range(num_layers)] + [ResBlock(model_channels, model_channels, dropout, dims=1, use_scale_shift_norm=True) for _ in range(3)])