holm/easydiffusion
1
0
Fork 0
forked from extern/easydiffusion
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Don't use colorama

Browse Source
This commit is contained in:
cmdr2 2022-09-16 21:32:08 +05:30 committed by GitHub
parent ef1bbda49c
commit 108e516b80
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
ui/sd_internal/runtime.py
View File

@ -33,7 +33,7 @@ filename_regex = re.compile('[^a-zA-Z0-9]')
from . import Request, Response, Image as ResponseImage
import base64
from io import BytesIO
from colorama import Fore
#from colorama import Fore
# local
stop_processing = False
@ -464,7 +464,7 @@ def mk_img(req: Request):
del x_samples
print("memory_final = ", torch.cuda.memory_allocated() / 1e6)
print(Fore.GREEN + 'Task completed')
print('Task completed')
if req.stream_progress_updates:
yield json.dumps(res.json())