Fix for Colab download status bar

2022-08-19 12:05:22 -04:00 · 2022-08-19 12:05:22 -04:00 · 7eee21d674
parent cf422aa16e
commit 7eee21d674
1 changed files with 1 additions and 0 deletions
--- a/utils.py
+++ b/utils.py
@ -279,6 +279,7 @@ def aria2_hook(pretrained_model_name_or_path: str, force_download=False, cache_d
                    done = True
                    break
                if bar is None:
+                    print("setting up status bar for aria2 download")
                    bar = tqdm(total=total_length, desc=f"[aria2] Downloading model", unit="B", unit_scale=True, unit_divisor=1000, file=Send_to_socketio())
                visited = set()
                for x in r: