From 91be38cba3cef8d522e659fa1ffa5c648d214bf7 Mon Sep 17 00:00:00 2001 From: James Betker Date: Wed, 8 Jun 2022 11:54:46 -0600 Subject: [PATCH] . --- codes/models/audio/music/gpt_music.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/codes/models/audio/music/gpt_music.py b/codes/models/audio/music/gpt_music.py index ac05f1dd..3b213974 100644 --- a/codes/models/audio/music/gpt_music.py +++ b/codes/models/audio/music/gpt_music.py @@ -27,7 +27,7 @@ class ConditioningEncoder(nn.Module): def forward(self, x): h = checkpoint(self.init, x) - h = self.attn(h + h = self.attn(h) return h.mean(dim=2)