From 8ba68e05ec610751c9699f347323d9ba51d8fb07 Mon Sep 17 00:00:00 2001 From: ebolam Date: Fri, 19 Aug 2022 12:13:46 -0400 Subject: [PATCH] Aria2 Status Bar Download Fix --- utils.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/utils.py b/utils.py index 493ec382..7fd82072 100644 --- a/utils.py +++ b/utils.py @@ -176,9 +176,8 @@ from flask_socketio import emit class Send_to_socketio(object): def write(self, bar): time.sleep(0.01) - print("got bar data") try: - print("Bar data: {}".format(bar)) + print(bar) emit('from_server', {'cmd': 'model_load_status', 'data': bar.replace(" ", " ")}, broadcast=True) except: pass @@ -279,7 +278,6 @@ 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: