mirror of
https://github.com/KoboldAI/KoboldAI-Client.git
synced 2025-06-05 21:59:24 +02:00
Fix for Colab download status bar
This commit is contained in:
1
utils.py
1
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:
|
||||
|
Reference in New Issue
Block a user