From e738e164e47d6968f2aa7778625eb1965a2cf061 Mon Sep 17 00:00:00 2001 From: ebolam Date: Sat, 15 Oct 2022 18:14:48 -0400 Subject: [PATCH] Add fix for status bar when chat mode ends early --- koboldai_settings.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/koboldai_settings.py b/koboldai_settings.py index 5ffcb681..44ec84b0 100644 --- a/koboldai_settings.py +++ b/koboldai_settings.py @@ -675,7 +675,7 @@ class model_settings(settings): self.tqdm.reset(total=self.genamt) self.tqdm_progress = 0 else: - self.tqdm.update(1) + self.tqdm.update(value-self.tqdm.n) self.tqdm_progress = int(float(self.generated_tkns)/float(self.genamt)*100) if self.tqdm.format_dict['rate'] is not None: self.tqdm_rem_time = str(datetime.timedelta(seconds=int(float(self.genamt-self.generated_tkns)/self.tqdm.format_dict['rate'])))