diff --git a/audiocraft/models/lm.py b/audiocraft/models/lm.py index 3bd68109..f772fef8 100644 --- a/audiocraft/models/lm.py +++ b/audiocraft/models/lm.py @@ -476,7 +476,7 @@ def generate(self, B, K, T = prompt.shape start_offset = T - assert start_offset < 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