Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

progress bar improvements #132

Open
wants to merge 1 commit into
base: main
Choose a base branch
from

Conversation

JCBrouwer
Copy link

Hey @s9roll7, another one with some improvements to progress bars. Addresses the main comment in #121.

  • The main progress bar now counts diffusion steps. I think this is easier to understand than the current steps per context window count.
  • Decoding latents with the VAE now also has its own progress bar (as especially for long / hi-res videos this can take quite a bit of time!)
  • I had to remove some info logs so that we don't get progress bars all over the place (rich/tqdm/logging being used all at once is really kind of a clusterfuck, so there's no better solution than this that I could find)

Here's what it looks like (running with a preview output every 5 steps):

new.progress.bar2.mp4

@SoftologyPro would this address your issue?

@JCBrouwer
Copy link
Author

hey, @s9roll7 just checking in, any feedback on this update to progress bars?

@@ -1148,9 +1146,6 @@ def preview_callback(i: int, video: torch.Tensor, save_fn: Callable[[torch.Tenso

seed_everything(seed)

logger.info(f"{len( region_condi_list )=}")
logger.info(f"{len( region_list )=}")

pipeline_output = pipeline(
Copy link
Owner

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Basically, don't delete the logs left in the code.
Most of the time, they are still being debugged or are left to investigate behavior that could be improved.

@@ -2886,6 +2898,7 @@ def sample_to_device( sample ):
tmp_latent = None

stopwatch_stop("LOOP end")
progress_bar.update()

Copy link
Owner

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

I do not want to change the units of progress

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants