fix sampling steps ui better

2023-07-02 17:03:06 +03:00 · 2023-07-02 17:03:06 +03:00 · 716efc72ca
parent dbb5fde6a8
commit 716efc72ca
1 changed files with 1 additions and 1 deletions
--- a/scripts/samplers/samplers_common.py
+++ b/scripts/samplers/samplers_common.py
@ -31,10 +31,10 @@ class SamplerStepCallback(object):
        self.total_steps = total_steps
        self.current_step = 0
        self.progress_bar = tqdm(desc=self.progress_msg(sampler_name, total_steps), total=total_steps)
-        state.sampling_steps = total_steps

    def progress_msg(self, name, total_steps=None):
        total_steps = total_steps if total_steps is not None else self.total_steps
+        state.sampling_steps = total_steps
        return f"Sampling Using {name} for {total_steps} steps."

    def set_webui_step(self, step):