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']}"