Surn commited on
Commit
618d8b2
·
1 Parent(s): 36c3689

Fix /0 bug

Browse files
Files changed (1) hide show
  1. audiocraft/models/musicgen.py +2 -2
audiocraft/models/musicgen.py CHANGED
@@ -411,8 +411,8 @@ class MusicGen:
411
 
412
  def _progress_callback(generated_tokens: int, tokens_to_generate: int):
413
  generated_tokens += current_gen_offset
414
- generated_tokens /= ((tokens_to_generate - 3) / self.duration)
415
- tokens_to_generate /= ((tokens_to_generate - 3) / self.duration)
416
  if self._progress_callback is not None:
417
  # Note that total_gen_len might be quite wrong depending on the
418
  # codebook pattern used, but with delay it is almost accurate.
 
411
 
412
  def _progress_callback(generated_tokens: int, tokens_to_generate: int):
413
  generated_tokens += current_gen_offset
414
+ generated_tokens /= ((tokens_to_generate) / self.duration)
415
+ tokens_to_generate /= ((tokens_to_generate) / self.duration)
416
  if self._progress_callback is not None:
417
  # Note that total_gen_len might be quite wrong depending on the
418
  # codebook pattern used, but with delay it is almost accurate.