mirror of
https://github.com/easydiffusion/easydiffusion.git
synced 2025-02-16 18:32:25 +01:00
Fix a bug where the task and req data needed to print with a backslash
This commit is contained in:
parent
dc6d48580b
commit
083390da83
@ -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)
|
||||
|
Loading…
Reference in New Issue
Block a user