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

also some good assert text

Browse Source
This commit is contained in:
James Betker 2022-07-14 21:26:22 -06:00
parent 4d53c66602
commit fa352e2744
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/tfdpc_v5.py
View File

@ -210,7 +210,7 @@ class TransformerDiffusionWithPointConditioning(nn.Module):
conditioning_input[:,:,tstart:tstart+tclip] = 0
if cond_left is None and self.new_cond:
assert cond_start > 20 and (cond_start+N+20 <= conditioning_input.shape[-1])
assert cond_start > 20 and (cond_start+N+20 <= conditioning_input.shape[-1]), f'{cond_start}, {N}, {conditioning_input.shape}'
cond_left = conditioning_input[:,:,:cond_start]
left_pt = -1
cond_right = conditioning_input[:,:,cond_start+N:]