From 9763c6bbe8747ab09f962e49c408d72554fa6b52 Mon Sep 17 00:00:00 2001 From: Vladimir Mandic Date: Wed, 8 Jan 2025 08:22:53 -0500 Subject: [PATCH] fix pbar error Signed-off-by: Vladimir Mandic --- extensions-builtin/sd-webui-agent-scheduler | 2 +- modules/processing_args.py | 5 ++++- 2 files changed, 5 insertions(+), 2 deletions(-) diff --git a/extensions-builtin/sd-webui-agent-scheduler b/extensions-builtin/sd-webui-agent-scheduler index 198e1b498..cd878626f 160000 --- a/extensions-builtin/sd-webui-agent-scheduler +++ b/extensions-builtin/sd-webui-agent-scheduler @@ -1 +1 @@ -Subproject commit 198e1b4980ab595468198f1d012f357ff4a31186 +Subproject commit cd878626f3b4f9a0c7c45c7d70b73a6168f612a4 diff --git a/modules/processing_args.py b/modules/processing_args.py index 2bc1c1ce2..e73cda85c 100644 --- a/modules/processing_args.py +++ b/modules/processing_args.py @@ -108,7 +108,10 @@ def set_pipeline_args(p, model, prompts:list, negative_prompts:list, prompts_2:t shared.sd_model = sd_models.apply_balanced_offload(shared.sd_model) apply_circular(p.tiling, model) if hasattr(model, "set_progress_bar_config"): - model.set_progress_bar_config(bar_format='Progress {rate_fmt}{postfix} {bar} {percentage:3.0f}% {n_fmt}/{total_fmt} {elapsed} {remaining} ' + '\x1b[38;5;71m' + desc, ncols=80, colour='#327fba', disable=disable_pbar) + if disable_pbar: + model.set_progress_bar_config(bar_format='Progress {rate_fmt}{postfix} {bar} {percentage:3.0f}% {n_fmt}/{total_fmt} {elapsed} {remaining} ' + '\x1b[38;5;71m' + desc, ncols=80, colour='#327fba', disable=disable_pbar) + else: + model.set_progress_bar_config(bar_format='Progress {rate_fmt}{postfix} {bar} {percentage:3.0f}% {n_fmt}/{total_fmt} {elapsed} {remaining} ' + '\x1b[38;5;71m' + desc, ncols=80, colour='#327fba') args = {} has_vae = hasattr(model, 'vae') or (hasattr(model, 'pipe') and hasattr(model.pipe, 'vae')) if hasattr(model, 'pipe') and not hasattr(model, 'no_recurse'): # recurse