Colab Fix

This commit is contained in:
ebolam
2022-09-19 09:37:31 -04:00
parent cd9e4175a9
commit 2d058afe3b

View File

@@ -224,6 +224,7 @@ def _download_with_aria2(aria2_config: str, total_length: int, directory: str =
if k not in visited: if k not in visited:
lengths[k] = (v[1], v[1]) 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())
koboldai_vars.total_download_chunks = sum(v[1] for v in lengths.values())
bar.n = koboldai_vars.downloaded_chunks bar.n = koboldai_vars.downloaded_chunks
bar.update() bar.update()
time.sleep(0.1) time.sleep(0.1)