rixty/KoboldAI-Client
1
1
Fork 0
mirror of https://github.com/KoboldAI/KoboldAI-Client.git synced 2025-06-05 21:59:24 +02:00
Code Issues Projects Releases Wiki Activity

Colab Fix

Browse Source
This commit is contained in:
ebolam
2022-09-19 09:45:38 -04:00
parent 5bfe6fd73f
commit 4d8df761ba
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
utils.py
View File

@@ -212,6 +212,7 @@ def _download_with_aria2(aria2_config: str, total_length: int, directory: str =
p.terminate()
done = True
break
visited = set()
for x in r:
filename = x["files"][0]["path"]
lengths[filename] = (int(x["completedLength"]), int(x["totalLength"]))
@@ -222,7 +223,6 @@ def _download_with_aria2(aria2_config: str, total_length: int, directory: str =
if bar is None:
bar = tqdm(total=total_length, desc=f"[aria2] Downloading model", unit="B", unit_scale=True, unit_divisor=1000)
koboldai_vars.total_download_chunks = sum(v[1] for v in lengths.values())
visited = set()
koboldai_vars.downloaded_chunks = sum(v[0] for v in lengths.values())
bar.n = koboldai_vars.downloaded_chunks
bar.update()