From fa352e2744418d0dc5c5f509ecb3c229e1412f15 Mon Sep 17 00:00:00 2001 From: James Betker Date: Thu, 14 Jul 2022 21:26:22 -0600 Subject: [PATCH] also some good assert text --- codes/models/audio/music/tfdpc_v5.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/codes/models/audio/music/tfdpc_v5.py b/codes/models/audio/music/tfdpc_v5.py index 96f25251..97a78172 100644 --- a/codes/models/audio/music/tfdpc_v5.py +++ b/codes/models/audio/music/tfdpc_v5.py @@ -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:]