Add options to disable/hide progress bar
See original GitHub issueIs your feature request related to a problem? Please describe.
This is a minor thing, but I find the progress bar annoying when I run inference with pipeline successively.
See this screenshot for example.
In this case, I generated 10 images using DDIMPipeline and used tqdm
myself, but the progress bars coming from __call__
of the pipeline are stacking up and annoying.
Describe the solution you’d like
It would be nice if disable
and leave
options of tqdm
were available with pipelines. Keyword arguments (say, disable_tqdm
and leave_tqdm
?) could be added to __call__
methods and passed to tqdm
. These are the relevant lines in case of DDPMPipeline
:
https://github.com/huggingface/diffusers/blob/92b6dbba1a25ed27b0bae38b089715132c7e6bcc/src/diffusers/pipelines/ddpm/pipeline_ddpm.py#L31
https://github.com/huggingface/diffusers/blob/92b6dbba1a25ed27b0bae38b089715132c7e6bcc/src/diffusers/pipelines/ddpm/pipeline_ddpm.py#L47
I found something that might be relevant in the logging module, but it’s not used in the pipeline modules. Maybe this can be used.
Issue Analytics
- State:
- Created a year ago
- Comments:7 (7 by maintainers)
Top GitHub Comments
Sure. I’ll look into it and make a PR.
Let’s also make sure that we can handle additional arguments to
tqdm
(liketotal=
) with**kwargs
like so: