diff --git a/workers/comfyui-json/data_types.py b/workers/comfyui-json/data_types.py index e73163d..fd8c6e5 100644 --- a/workers/comfyui-json/data_types.py +++ b/workers/comfyui-json/data_types.py @@ -13,8 +13,9 @@ with open("workers/comfyui/misc/test_prompts.txt", "r") as f: test_prompts = f.readlines() def count_workload() -> float: - # Always 1.0 where there is a single instance of ComfyUI handling requests - return 1.0 + # Always 100.0 where there is a single instance of ComfyUI handling requests + # Results will indicate % or a job completed per second. Avoids sub 0.1 sec performance indication + return 100.0 @dataclasses.dataclass class ComfyWorkflowData(ApiPayload): diff --git a/workers/comfyui-json/server.py b/workers/comfyui-json/server.py index 702330e..44b10fc 100644 --- a/workers/comfyui-json/server.py +++ b/workers/comfyui-json/server.py @@ -70,7 +70,7 @@ class ComfyWorkflowHandler(EndpointHandler[ComfyWorkflowData]): @property def healthcheck_endpoint(self) -> Optional[str]: - return None + return "/health" @classmethod def payload_cls(cls) -> Type[ComfyWorkflowData]: