From 40023ee975417695774613f193381b98a4f6852f Mon Sep 17 00:00:00 2001 From: Kevin Sasso Date: Sun, 28 Apr 2024 09:31:50 -0700 Subject: [PATCH 1/2] Update blocks.py Allow Unet to process any arbitrary size (above kernel size) --- audio_diffusion/blocks.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/audio_diffusion/blocks.py b/audio_diffusion/blocks.py index 6261a90..803c630 100644 --- a/audio_diffusion/blocks.py +++ b/audio_diffusion/blocks.py @@ -52,7 +52,10 @@ def __init__(self, *main): self.main = nn.Sequential(*main) def forward(self, input): - return torch.cat([self.main(input), input], dim=1) + main = self.main(input) + pad = (0, (input.size(-1) - main.size(-1))) + main = torch.nn.functional.pad(main, pad) + return main class FourierFeatures(nn.Module): def __init__(self, in_features, out_features, std=1.): From 2a3d474250b10dc1d39c373be5dbb307361b8698 Mon Sep 17 00:00:00 2001 From: kevin sasso Date: Sun, 28 Apr 2024 17:56:58 -0700 Subject: [PATCH 2/2] lol removed concat by accident --- audio_diffusion/blocks.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/audio_diffusion/blocks.py b/audio_diffusion/blocks.py index 803c630..a4fbf05 100644 --- a/audio_diffusion/blocks.py +++ b/audio_diffusion/blocks.py @@ -55,7 +55,7 @@ def forward(self, input): main = self.main(input) pad = (0, (input.size(-1) - main.size(-1))) main = torch.nn.functional.pad(main, pad) - return main + return torch.cat([main, input], dim=1) class FourierFeatures(nn.Module): def __init__(self, in_features, out_features, std=1.):