fix: avoid crashing while recording jobs when using no loras or TI

Haidra-Org · Jul 21, 2024 · 80816f0 · 80816f0
1 parent 726978b
commit 80816f0
Showing 1 changed file with 8 additions and 6 deletions.
diff --git 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