Fix LM assert error

app.py

@@ -266,14 +266,14 @@ def predict(model, text, melody_filepath, duration, dimension, topk, topp, tempe
                     descriptions=[text],
                     melody_wavs=melody,
                     melody_sample_rate=sr,
-                    progress=True
+                    progress=False
                 )
             # All output_segments are populated, so we can break the loop or set duration to 0
             break
         else:
             #output = MODEL.generate(descriptions=[text], progress=False)
             if not output_segments:
-                next_segment = MODEL.generate(descriptions=[text], progress=True)
+                next_segment = MODEL.generate(descriptions=[text], progress=False)
                 duration -= segment_duration
             else:
                 last_chunk = output_segments[-1][:, :, -overlap*MODEL.sample_rate:]

audiocraft/models/lm.py

@@ -456,7 +456,8 @@ class LMModel(StreamingModule):
 
         B, K, T = prompt.shape
         start_offset = T
-        assert start_offset < max_gen_len
+        print(f"start_offset: {start_offset} | max_gen_len: {max_gen_len}")
+        assert start_offset <= max_gen_len
 
         pattern = self.pattern_provider.get_pattern(max_gen_len)
         # this token is used as default value for codes that are not generated yet