From e652c017a6dad75a935791fcad6fb5689e8db4e8 Mon Sep 17 00:00:00 2001 From: tazlin Date: Sun, 1 Oct 2023 12:23:10 -0400 Subject: [PATCH] fix: set denoise to `None` if no `source_image` --- horde_worker_regen/process_management/process_manager.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/horde_worker_regen/process_management/process_manager.py b/horde_worker_regen/process_management/process_manager.py index 2ba6ce84..9453556e 100644 --- a/horde_worker_regen/process_management/process_manager.py +++ b/horde_worker_regen/process_management/process_manager.py @@ -1116,6 +1116,12 @@ async def api_job_pop(self) -> None: new_response_dict["payload"]["seed"] = random.randint(0, (2**32) - 1) job_pop_response = ImageGenerateJobPopResponse(**new_response_dict) + if job_pop_response.payload.denoising_strength is not None and job_pop_response.source_image is None: + logger.warning(f"Job {job_pop_response.id_} has denoising_strength but no source image!") + new_response_dict = job_pop_response.model_dump() + new_response_dict["payload"]["denoising_strength"] = None + job_pop_response = ImageGenerateJobPopResponse(**new_response_dict) + if job_pop_response.source_image is not None and "https://" in job_pop_response.source_image: # Download and convert the source image to base64 fail_count = 0