Update audiocraft/models/musicgen.py

audiocraft/models/musicgen.py

@@ -323,7 +323,7 @@ class MusicGen:
             torch.Tensor: Generated audio, of shape [B, C, T], T is defined by the generation params.
         """
         total_gen_len = int(self.duration * self.frame_rate)
-        max_prompt_len = int(min(self.duration, self.max_duration) * self.frame_rate)
+        max_prompt_len = int(min(int(self.duration), int(self.max_duration)) * self.frame_rate)
         current_gen_offset: int = 0
 
         def _progress_callback(generated_tokens: int, tokens_to_generate: int):