fix(api): disable progress in diffusers pipelines

ssube · Feb 18, 2023 · b2de114 · b2de114
1 parent 8dd55cc
commit b2de114
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git 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
@@ -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())