Spaces:
Running
on
T4
Running
on
T4
Fix /0 bug
Browse files
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
|
415 |
-
tokens_to_generate /= ((tokens_to_generate
|
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.
|