crowsonkb / k-diffusion

Karras et al. (2022) diffusion models for PyTorch
MIT License
2.26k stars 372 forks source link

Req: TQDM trange() update frequency / miniters option #27

Open torridgristle opened 1 year ago

torridgristle commented 1 year ago

I'm having a problem in Colab where the cell won't be responsive and can't be stopped if left running for over 10 minutes, and because I can reload the page and suddenly see a lot of new images at once I think there's a backlog of cell outputs that weren't displayed yet and because they weren't displayed yet I couldn't stop the cell?

The only possible solution to this I can find is the miniters argument for TQDM, where a value above zero will skip that many iterations between updates to the output progress bar.

I don't know of a good way to implement this option that wouldn't just stick it in the sampler arguments, but maybe that would be fine?