diff --git a/utils.py b/utils.py index 0ac311db..10e2bffc 100644 --- a/utils.py +++ b/utils.py @@ -224,6 +224,7 @@ def _download_with_aria2(aria2_config: str, total_length: int, directory: str = 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.total_download_chunks = sum(v[1] for v in lengths.values()) bar.n = koboldai_vars.downloaded_chunks bar.update() time.sleep(0.1)