Spaces:
Sleeping
Sleeping
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.
|