diffusers icon indicating copy to clipboard operation
diffusers copied to clipboard

Can the progress bar from pipe be suppressed ?

Open Trisanu-007 opened this issue 2 years ago • 1 comments

Hi, When we pass a prompt to the pip (from for eg: pipe = StableDiffusionPipeline.from_pretrained("./stable-diffusion-v1-5")), it displays an output in this case, with a progress bar. Is it possible to get an output without the progress bar, or to somehow disable it?

Thanks in advance!

Trisanu-007 avatar Dec 20 '22 12:12 Trisanu-007

@Trisanu-007 👋 great question. Here is some information to help you out.

Diffusers Docs

diffusers.utils.logging.disable_progress_bar

https://huggingface.co/docs/diffusers/api/logging#diffusers.utils.logging.disable_progress_bar

Related Issues / Pull Requests

#172 - Add options to disable/hide progress bar

#242 - Refactor progress bar

pipeline = DDPMPipeline.from_pretrained('CompVis/ldm-celebahq-256')
pipeline.to('cuda:0')

# if one wants to set `leave=False`
pipeline.set_progress_bar_config(leave=False)

# if one wants to disable `tqdm`
pipeline.set_progress_bar_config(disable=True)

averad avatar Dec 20 '22 17:12 averad

Thank you very very much ! This solves my issue !

Trisanu-007 avatar Dec 21 '22 08:12 Trisanu-007