diff --git a/horde_worker_regen/process_management/process_manager.py b/horde_worker_regen/process_management/process_manager.py index 05e1d568..554b08dc 100644 --- a/horde_worker_regen/process_management/process_manager.py +++ b/horde_worker_regen/process_management/process_manager.py @@ -2699,12 +2699,14 @@ async def api_submit_job(self) -> None: else: model_dump["sdk_api_job_info"]["payload"]["scheduler"] = "simple" del model_dump["sdk_api_job_info"]["payload"]["karras"] - model_dump["sdk_api_job_info"]["payload"]["lora_count"] = len( - model_dump["sdk_api_job_info"]["payload"]["loras"], - ) - model_dump["sdk_api_job_info"]["payload"]["ti_count"] = len( - model_dump["sdk_api_job_info"]["payload"]["tis"], - ) + if model_dump["sdk_api_job_info"]["payload"]["loras"] is not None: + model_dump["sdk_api_job_info"]["payload"]["lora_count"] = len( + model_dump["sdk_api_job_info"]["payload"]["loras"], + ) + if model_dump["sdk_api_job_info"]["payload"]["tis"] is not None: + model_dump["sdk_api_job_info"]["payload"]["ti_count"] = len( + model_dump["sdk_api_job_info"]["payload"]["tis"], + ) model_dump["sdk_api_job_info"]["extra_source_images_count"] = ( len(hji.sdk_api_job_info.extra_source_images) if hji.sdk_api_job_info.extra_source_images