From 7244466c92b62912054451402e844a6f4600cd2e Mon Sep 17 00:00:00 2001 From: GeekyBecky Date: Wed, 16 Oct 2024 16:13:11 -0700 Subject: [PATCH] skips non-dicts and text outputs in comfyui_response --- .../opt/ai-dock/api-wrapper/workers/postprocess_worker.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/build/COPY_ROOT_1/opt/ai-dock/api-wrapper/workers/postprocess_worker.py b/build/COPY_ROOT_1/opt/ai-dock/api-wrapper/workers/postprocess_worker.py index a2e38b9c..96576a70 100644 --- a/build/COPY_ROOT_1/opt/ai-dock/api-wrapper/workers/postprocess_worker.py +++ b/build/COPY_ROOT_1/opt/ai-dock/api-wrapper/workers/postprocess_worker.py @@ -59,6 +59,8 @@ async def move_assets(self, request_id, result): for inner_key, inner_value in value.items(): if isinstance(inner_value, list): for item in inner_value: + if not isinstance(item, dict): + continue if item.get("type") == "output": original_path = f"{config.OUTPUT_DIR}{item['subfolder']}/{item['filename']}" new_path = f"{custom_output_dir}/{item['filename']}"