From b2de114c8c71679959ff7c9a93a69c1d17f34f2b Mon Sep 17 00:00:00 2001 From: Sean Sube Date: Sat, 18 Feb 2023 09:42:38 -0600 Subject: [PATCH] fix(api): disable progress in diffusers pipelines --- api/onnx_web/chain/upscale_stable_diffusion.py | 3 +++ api/onnx_web/diffusion/load.py | 3 +++ 2 files changed, 6 insertions(+) diff --git a/api/onnx_web/chain/upscale_stable_diffusion.py b/api/onnx_web/chain/upscale_stable_diffusion.py index e6580dd0..c2032bfb 100644 --- a/api/onnx_web/chain/upscale_stable_diffusion.py +++ b/api/onnx_web/chain/upscale_stable_diffusion.py @@ -49,6 +49,9 @@ def load_stable_diffusion( provider=device.provider, ) + if not server.show_progress: + pipe.set_progress_bar_config(disable=True) + server.cache.set("diffusion", cache_key, pipe) run_gc([device]) diff --git a/api/onnx_web/diffusion/load.py b/api/onnx_web/diffusion/load.py index 60072621..dceb509f 100644 --- a/api/onnx_web/diffusion/load.py +++ b/api/onnx_web/diffusion/load.py @@ -148,6 +148,9 @@ def load_pipeline( scheduler=scheduler, ) + if not server.show_progress: + pipe.set_progress_bar_config(disable=True) + if device is not None and hasattr(pipe, "to"): pipe = pipe.to(device.torch_str())