diff --git a/codes/models/audio/music/tfdpc_v5.py b/codes/models/audio/music/tfdpc_v5.py
index 69d96346..60a7674d 100644
--- a/codes/models/audio/music/tfdpc_v5.py
+++ b/codes/models/audio/music/tfdpc_v5.py
@@ -197,7 +197,7 @@ class TransformerDiffusionWithPointConditioning(nn.Module):
         if custom_conditioning_fetcher is not None:
             cs, ce = custom_conditioning_fetcher(self.conditioning_encoder, time_emb)
         else:
-            assert conditioning_input.shape[-1] - cond_start - N > 0, f'Some sort of conditioning misalignment, {conditioning_input.shape[-1], cond_start, N}'
+            assert conditioning_input.shape[-1] - cond_start - N >= 0, f'Some sort of conditioning misalignment, {conditioning_input.shape[-1], cond_start, N}'
             cond_pre = conditioning_input[:,:,:cond_start]
             cond_aligned = conditioning_input[:,:,cond_start:N+cond_start]
             cond_post = conditioning_input[:,:,N+cond_start:]
@@ -308,7 +308,7 @@ def inference_tfdpc5_with_cheater():
     with torch.no_grad():
         os.makedirs('results/tfdpc_v3', exist_ok=True)
 
-        #length = 40 * 22050 // 256 // 16
+        # length = 40 * 22050 // 256 // 16
         samples = {'electronica1': load_audio('Y:\\split\\yt-music-eval\\00001.wav', 22050),
                    'electronica2': load_audio('Y:\\split\\yt-music-eval\\00272.wav', 22050),
                    'e_guitar': load_audio('Y:\\split\\yt-music-eval\\00227.wav', 22050),