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())