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

Fix another edge case

Browse Source
This commit is contained in:
James Betker 2022-07-04 16:47:57 -06:00
parent 808a1a4a31
commit 802998674e
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

@ -203,7 +203,7 @@ class TransformerDiffusionWithPointConditioning(nn.Module):
else:
if self.training and self.conditioning_masking > 0:
mask_prop = random.random() * self.conditioning_masking
mask_len = max(int(N * mask_prop), 4)
mask_len = max(int(N * mask_prop), 16)
assert N-mask_len > 8, f"Use longer inputs or shorter conditioning_masking proportion. {N-mask_len}"
seg_start = random.randint(8, (N-mask_len)) + cond_start
# Readjust mask_len to ensure at least 8 sequence elements on the end as well