From 083390da83ec2d986835fbf8010baa5c77cf5e6f Mon Sep 17 00:00:00 2001 From: cmdr2 Date: Wed, 28 Dec 2022 19:23:36 +0530 Subject: [PATCH] Fix a bug where the task and req data needed to print with a backslash --- ui/easydiffusion/renderer.py | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) diff --git a/ui/easydiffusion/renderer.py b/ui/easydiffusion/renderer.py index 478b7284..852cc0a2 100644 --- a/ui/easydiffusion/renderer.py +++ b/ui/easydiffusion/renderer.py @@ -29,8 +29,7 @@ def init(device): def make_images(req: GenerateImageRequest, task_data: TaskData, data_queue: queue.Queue, task_temp_images: list, step_callback): context.stop_processing = False - log.info(f'request: {pprint.pformat(get_printable_request(req)).replace("[","\[")}') - log.info(f'task data: {pprint.pformat(task_data.dict()).replace("[","\[")}') + print_task_info(req, task_data) images = make_images_internal(req, task_data, data_queue, task_temp_images, step_callback) @@ -41,6 +40,12 @@ def make_images(req: GenerateImageRequest, task_data: TaskData, data_queue: queu return res +def print_task_info(req: GenerateImageRequest, task_data: TaskData): + req_str = pprint.pformat(get_printable_request(req)).replace("[","\[") + task_str = pprint.pformat(task_data.dict()).replace("[","\[") + log.info(f'request: {req_str}') + log.info(f'task data: {task_str}') + def make_images_internal(req: GenerateImageRequest, task_data: TaskData, data_queue: queue.Queue, task_temp_images: list, step_callback): images, user_stopped = generate_images_internal(req, task_data, data_queue, task_temp_images, step_callback, task_data.stream_image_progress) filtered_images = filter_images(task_data, images, user_stopped)