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

fix obo

Browse Source
This commit is contained in:
James Betker 2022-07-12 22:28:20 -06:00
parent f46d6645da
commit ebfe72d502
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

@ -212,7 +212,7 @@ class TransformerDiffusionWithPointConditioning(nn.Module):
if cond_left is None and self.new_cond: if cond_left is None and self.new_cond:
cond_left = conditioning_input[:,:,:max(cond_start, 20)] cond_left = conditioning_input[:,:,:max(cond_start, 20)]
left_pt = cond_start left_pt = cond_start-1
cond_right = conditioning_input[:,:,min(N+cond_start, conditioning_input.shape[-1]-20):] cond_right = conditioning_input[:,:,min(N+cond_start, conditioning_input.shape[-1]-20):]
right_pt = cond_right.shape[-1] - (conditioning_input.shape[-1] - (N+cond_start)) right_pt = cond_right.shape[-1] - (conditioning_input.shape[-1] - (N+cond_start))
elif cond_left is None: elif cond_left is None: