From f5ce00622b85be7adee8cb8656db2573103e20e7 Mon Sep 17 00:00:00 2001 From: ebolam Date: Mon, 19 Sep 2022 09:21:08 -0400 Subject: [PATCH] Fix for UI2 download status bar --- aiserver.py | 2 +- utils.py | 3 ++- 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/aiserver.py b/aiserver.py index 1fe206af..b594dbc0 100644 --- a/aiserver.py +++ b/aiserver.py @@ -1674,7 +1674,7 @@ def patch_transformers_download(): if bar != "": try: - print(bar, end="") + print(bar, end="\n") emit('from_server', {'cmd': 'model_load_status', 'data': bar.replace(" ", " ")}, broadcast=True, room="UI_1") eventlet.sleep(seconds=0) except: diff --git a/utils.py b/utils.py index d4af0b65..661d2bbd 100644 --- a/utils.py +++ b/utils.py @@ -207,6 +207,7 @@ def _download_with_aria2(aria2_config: str, total_length: int, directory: str = if bar is not None: bar.n = bar.total bar.close() + koboldai_vars.downloaded_chunks = total p.terminate() done = True break @@ -221,7 +222,7 @@ def _download_with_aria2(aria2_config: str, total_length: int, directory: str = for k, v in lengths.items(): if k not in visited: lengths[k] = (v[1], v[1]) - koboldai_vars.downloaded_chunks = sum(v[0] for v in lengths.values()) + koboldai_vars.downloaded_chunks += sum(v[0] for v in lengths.values()) bar.n = koboldai_vars.downloaded_chunks bar.update() time.sleep(0.1)