Fix bug in non-melody music generation length

audiocraft/models/musicgen.py

@@ -424,7 +424,7 @@ class MusicGen:
                 print(f'{generated_tokens: 6.2f} / {tokens_to_generate: 6.2f}', end='\r')
 
         if prompt_tokens is not None:
-            assert max_prompt_len >= prompt_tokens.shape[-1], \
+            assert max_prompt_len > prompt_tokens.shape[-1], \
                 "Prompt is longer than audio to generate"
 
         # callback = None