diff --git a/utils.py b/utils.py index 91fc5abc..84984f32 100644 --- a/utils.py +++ b/utils.py @@ -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()