From ebfe72d502e1de0483302d6ecb77c0f5fd47ed0a Mon Sep 17 00:00:00 2001 From: James Betker Date: Tue, 12 Jul 2022 22:28:20 -0600 Subject: [PATCH] fix obo --- 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 7efe1a3b..38c0a09b 100644 --- a/codes/models/audio/music/tfdpc_v5.py +++ b/codes/models/audio/music/tfdpc_v5.py @@ -212,7 +212,7 @@ class TransformerDiffusionWithPointConditioning(nn.Module): if cond_left is None and self.new_cond: 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):] right_pt = cond_right.shape[-1] - (conditioning_input.shape[-1] - (N+cond_start)) elif cond_left is None: