mrq/DL-Art-School
1
3
Fork 5
Code Issues Pull Requests Packages Projects Releases Wiki Activity

.

Browse Source
This commit is contained in:
James Betker 2022-06-08 11:54:46 -06:00
parent dee2b72786
commit 91be38cba3
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
codes/models/audio/music/gpt_music.py
View File

@ -27,7 +27,7 @@ class ConditioningEncoder(nn.Module):
def forward(self, x): def forward(self, x):
h = checkpoint(self.init, x) h = checkpoint(self.init, x)
h = self.attn(h h = self.attn(h)
return h.mean(dim=2) return h.mean(dim=2)