rixty
/
KoboldAI-Client
mirror of https://github.com/KoboldAI/KoboldAI-Client.git
1
1
Fork 0
Code Issues Projects Releases Wiki Activity

Merge pull request #174 from ebolam/united 

Fix for blank model info box when downloading model
This commit is contained in:
henk717 2022-07-29 22:15:58 +02:00 committed by GitHub
parent 699c2353e7 f97c10b794
commit a63f7cfa5a
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 8 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
aiserver.py
View File

@ -1319,13 +1319,14 @@ def patch_transformers_download():
import copy, requests, tqdm, time import copy, requests, tqdm, time
class Send_to_socketio(object): class Send_to_socketio(object):
def write(self, bar): def write(self, bar):
bar = bar.replace("\r", "") bar = bar.replace("\r", "").replace("\n", "")
try: if bar != "":
print(bar, end="\r") try:
emit('from_server', {'cmd': 'model_load_status', 'data': bar.replace(" ", " ")}, broadcast=True) print(bar, end="\r")
eventlet.sleep(seconds=0) emit('from_server', {'cmd': 'model_load_status', 'data': bar.replace(" ", " ")}, broadcast=True)
except: eventlet.sleep(seconds=0)
pass except:
pass
def http_get( def http_get(
url: str, url: str,
temp_file: transformers.utils.hub.BinaryIO, temp_file: transformers.utils.hub.BinaryIO,