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 UI2 download status bar

Browse Source
This commit is contained in:
ebolam
2022-09-19 09:21:08 -04:00
parent ca3db6a05c
commit f5ce00622b
2 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
aiserver.py
View File

@@ -1674,7 +1674,7 @@ def patch_transformers_download():
if bar != "":
try:
print(bar, end="")
print(bar, end="\n")
emit('from_server', {'cmd': 'model_load_status', 'data': bar.replace(" ", " ")}, broadcast=True, room="UI_1")
eventlet.sleep(seconds=0)
except:

3
utils.py
View File

@@ -207,6 +207,7 @@ def _download_with_aria2(aria2_config: str, total_length: int, directory: str =
if bar is not None:
bar.n = bar.total
bar.close()
koboldai_vars.downloaded_chunks = total
p.terminate()
done = True
break
@@ -221,7 +222,7 @@ def _download_with_aria2(aria2_config: str, total_length: int, directory: str =
for k, v in lengths.items():
if k not in visited:
lengths[k] = (v[1], v[1])
koboldai_vars.downloaded_chunks = sum(v[0] for v in lengths.values())
koboldai_vars.downloaded_chunks += sum(v[0] for v in lengths.values())
bar.n = koboldai_vars.downloaded_chunks
bar.update()
time.sleep(0.1)