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-07-14 21:52:23 -06:00
parent 51291ab070
commit e13b1adfdb
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
codes/models/audio/music/tfdpc_v5.py
View File

@ -219,9 +219,9 @@ class TransformerDiffusionWithPointConditioning(nn.Module):
if self.training: if self.training:
# Arbitrarily restrict the context given. We should support short contexts and without this they are never encountered. # Arbitrarily restrict the context given. We should support short contexts and without this they are never encountered.
arb_context_cap = random.randint(50, 100) arb_context_cap = random.randint(50, 100)
if cond_left.shape[-1] > arb_context_cap and random() > .5: if cond_left.shape[-1] > arb_context_cap and random.random() > .5:
cond_left = cond_left[:,:,-arb_context_cap:] cond_left = cond_left[:,:,-arb_context_cap:]
if cond_right.shape[-1] > arb_context_cap and random() > .5: if cond_right.shape[-1] > arb_context_cap and random.random() > .5:
cond_right = cond_right[:,:,:arb_context_cap] cond_right = cond_right[:,:,:arb_context_cap]
elif cond_left is None: elif cond_left is None: