extern/easydiffusion
1
0
Fork 1
mirror of https://github.com/easydiffusion/easydiffusion.git synced 2025-04-24 11:28:19 +02:00
Code Issues 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 from . import Request, Response, Image as ResponseImage
import base64 import base64
from io import BytesIO from io import BytesIO
from colorama import Fore #from colorama import Fore
# local # local
stop_processing = False stop_processing = False
@ -464,7 +464,7 @@ def mk_img(req: Request):
del x_samples del x_samples
print("memory_final = ", torch.cuda.memory_allocated() / 1e6) print("memory_final = ", torch.cuda.memory_allocated() / 1e6)
print(Fore.GREEN + 'Task completed') print('Task completed')
if req.stream_progress_updates: if req.stream_progress_updates:
yield json.dumps(res.json()) yield json.dumps(res.json())