From ae9ebba185620d44be5ce96c70712f712b039679 Mon Sep 17 00:00:00 2001 From: ebolam Date: Sat, 22 Oct 2022 21:09:45 -0400 Subject: [PATCH] Fix for Model Download Bar on UI2 --- koboldai_settings.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/koboldai_settings.py b/koboldai_settings.py index 3a60a467..23fdd82b 100644 --- a/koboldai_settings.py +++ b/koboldai_settings.py @@ -691,7 +691,12 @@ class model_settings(settings): else: self.tqdm.update(value-old_value) if self.total_download_chunks is not None: - self.tqdm_progress = 0 if self.total_download_chunks==0 else round(float(self.downloaded_chunks)/float(self.total_download_chunks)*100, 1) + if self.total_download_chunks==0: + self.tqdm_progress = 0 + elif float(self.downloaded_chunks) > float(self.total_download_chunks): + self.tqdm_progress = 100 + else: + self.tqdm_progress = round(float(self.downloaded_chunks)/float(self.total_download_chunks)*100, 1) else: self.tqdm_progress = 0 if self.tqdm.format_dict['rate'] is not None: