From 4d8df761bac64ece8c8d530093c70c928cc04297 Mon Sep 17 00:00:00 2001
From: ebolam <ebolam@gmail.com>
Date: Mon, 19 Sep 2022 09:45:38 -0400
Subject: [PATCH] Colab Fix

---
 utils.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

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()