extern/easydiffusion
1
0
Fork 1
mirror of https://github.com/easydiffusion/easydiffusion.git synced 2025-04-29 22:04:25 +02:00
Code Issues Packages Projects Releases Wiki Activity

Fix a bug where the task and req data needed to print with a backslash

Browse Source
This commit is contained in:
cmdr2 2022-12-28 19:23:36 +05:30
parent dc6d48580b
commit 083390da83
1 changed files with 7 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
ui/easydiffusion/renderer.py
View File

@ -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): def make_images(req: GenerateImageRequest, task_data: TaskData, data_queue: queue.Queue, task_temp_images: list, step_callback):
context.stop_processing = False context.stop_processing = False
log.info(f'request: {pprint.pformat(get_printable_request(req)).replace("[","\[")}') print_task_info(req, task_data)
log.info(f'task data: {pprint.pformat(task_data.dict()).replace("[","\[")}')
images = make_images_internal(req, task_data, data_queue, task_temp_images, step_callback) 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 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): 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) 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) filtered_images = filter_images(task_data, images, user_stopped)