From 108e516b8049dfbec75567bb6a8a65a828e614b5 Mon Sep 17 00:00:00 2001 From: cmdr2 Date: Fri, 16 Sep 2022 21:32:08 +0530 Subject: [PATCH] Don't use colorama --- ui/sd_internal/runtime.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/ui/sd_internal/runtime.py b/ui/sd_internal/runtime.py index 04b59d32..c05dde45 100644 --- a/ui/sd_internal/runtime.py +++ b/ui/sd_internal/runtime.py @@ -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())