rixty/KoboldAI-Client
1
1
Fork 0
mirror of https://github.com/KoboldAI/KoboldAI-Client.git synced 2025-06-05 21:59:24 +02:00
Code Issues Projects Releases Wiki Activity

Fix for aria2 download status to UI

Browse Source
This commit is contained in:
ebolam
2022-09-21 13:07:49 -04:00
parent f3d6beb578
commit 0e270e0b25
1 changed files with 15 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
utils.py
View File

@@ -188,6 +188,20 @@ class Send_to_socketio(object):
pass pass
def _download_with_aria2(aria2_config: str, total_length: int, directory: str = ".", user_agent=None, force_download=False, use_auth_token=None): def _download_with_aria2(aria2_config: str, total_length: int, directory: str = ".", user_agent=None, force_download=False, use_auth_token=None):
class Send_to_socketio(object):
def write(self, bar):
bar = bar.replace("\r", "").replace("\n", "")
if bar != "":
try:
print(bar, end="\n")
emit('from_server', {'cmd': 'model_load_status', 'data': bar.replace(" ", " ")}, broadcast=True)
eventlet.sleep(seconds=0)
except:
pass
def flush(self):
pass
import transformers import transformers
lengths = {} lengths = {}
s = requests.Session() s = requests.Session()
@@ -211,7 +225,7 @@ def _download_with_aria2(aria2_config: str, total_length: int, directory: str =
done = True done = True
break break
if bar is None: if bar is None:
bar = 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, file=Send_to_socketio())
visited = set() visited = set()
for x in r: for x in r:
filename = x["files"][0]["path"] filename = x["files"][0]["path"]