Colab Fix

This commit is contained in:
ebolam
2022-09-19 09:30:30 -04:00
parent d1c56bc6b8
commit 48101f492d

View File

@@ -213,7 +213,7 @@ def _download_with_aria2(aria2_config: str, total_length: int, directory: str =
done = True
break
if bar is None:
bar = tqdm.tqdm(total=total_length, desc=f"[aria2] Downloading model", unit="B", unit_scale=True, unit_divisor=1000)
bar = tqdm(total=total_length, desc=f"[aria2] Downloading model", unit="B", unit_scale=True, unit_divisor=1000)
koboldai_vars.total_download_chunks = total_length
visited = set()
for x in r: