Merge pull request #174 from ebolam/united
Fix for blank model info box when downloading model
This commit is contained in:
commit
a63f7cfa5a
15
aiserver.py
15
aiserver.py
|
@ -1319,13 +1319,14 @@ def patch_transformers_download():
|
|||
import copy, requests, tqdm, time
|
||||
class Send_to_socketio(object):
|
||||
def write(self, bar):
|
||||
bar = bar.replace("\r", "")
|
||||
try:
|
||||
print(bar, end="\r")
|
||||
emit('from_server', {'cmd': 'model_load_status', 'data': bar.replace(" ", " ")}, broadcast=True)
|
||||
eventlet.sleep(seconds=0)
|
||||
except:
|
||||
pass
|
||||
bar = bar.replace("\r", "").replace("\n", "")
|
||||
if bar != "":
|
||||
try:
|
||||
print(bar, end="\r")
|
||||
emit('from_server', {'cmd': 'model_load_status', 'data': bar.replace(" ", " ")}, broadcast=True)
|
||||
eventlet.sleep(seconds=0)
|
||||
except:
|
||||
pass
|
||||
def http_get(
|
||||
url: str,
|
||||
temp_file: transformers.utils.hub.BinaryIO,
|
||||
|
|
Loading…
Reference in New Issue