feat: add DISCOART_DISABLE_TQDM to control tqdm

jina-ai · Aug 4, 2022 · 48bb8fa · 48bb8fa
1 parent 53a8a36
commit 48bb8fa
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/README.md b/README.md
@@ -296,6 +296,7 @@ DISCOART_CACHE_DIR='path/to/your-cache-dir' # use a custom cache directory for m
 DISCOART_DISABLE_REMOTE_MODELS='1' # disable the listing of diffusion models on Github, remote diffusion models allows user to use latest models without updating the codebase.
 DISCOART_REMOTE_MODELS_URL='https://yourdomain/models.yml' # use a custom remote URL for fetching models list
 DISCOART_DISABLE_CHECK_MODEL_SHA='1' # disable checking local model SHA matches the remote model SHA
+DISCOART_DISABLE_TQDM='1' # disable tqdm progress bar on diffusion
 ```
 
 ## CLI

diff --git a/discoart/runner.py b/discoart/runner.py
@@ -380,7 +380,7 @@ def cond_fn(x, t, **kwargs):
                 clip_denoised=args.clip_denoised,
                 model_kwargs={},
                 cond_fn=cond_fn,
-                progress=_is_jupyter,
+                progress='DISCOART_DISABLE_TQDM' not in os.environ,
                 skip_timesteps=skip_steps,
                 init_image=init,
                 randomize_class=args.randomize_class,
@@ -397,7 +397,7 @@ def cond_fn(x, t, **kwargs):
                 clip_denoised=args.clip_denoised,
                 model_kwargs={},
                 cond_fn=cond_fn,
-                progress=_is_jupyter,
+                progress='DISCOART_DISABLE_TQDM' not in os.environ,
                 skip_timesteps=skip_steps,
                 init_image=init,
                 randomize_class=args.randomize_class,