1movmedia/pyworker
2
0
Fork 0
Code Pull Requests Activity

Merge pull request #4 from robballantyne/feat/comfyui-json-benchmark-workflow-from-file-no-silent-fail

Browse Source 
Feat/comfyui json benchmark workflow from file no silent fail
This commit is contained in:
Rob Ballantyne
2025-10-23 17:02:12 +01:00
committed by GitHub
parent ec25dda3ad f4f7080df1
commit 63909736bb
1 changed files with 5 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
workers/comfyui-json/data_types.py
+5 -1
View File
@@ -7,9 +7,12 @@ from functools import cache
from math import ceil from math import ceil
from pathlib import Path from pathlib import Path
import json import json
import logging
from lib.data_types import ApiPayload, JsonDataException from lib.data_types import ApiPayload, JsonDataException
log = logging.getLogger(__file__)
def count_workload() -> float: def count_workload() -> float:
# Always 100.0 where there is a single instance of ComfyUI handling requests # 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 # Results will indicate % or a job completed per second. Avoids sub 0.1 sec performance indication
@@ -41,9 +44,10 @@ class ComfyWorkflowData(ApiPayload):
) )
except (json.JSONDecodeError, IOError): except (json.JSONDecodeError, IOError):
# JSON is malformed or file can't be read, fall through to default # JSON is malformed or file can't be read, fall through to default
pass log.error(f"Failed to benchmark using {benchmark_file}")
# Fallback: read prompts and construct payload # Fallback: read prompts and construct payload
log.info("Using fallback method for benchmarking")
with open("workers/comfyui-json/misc/test_prompts.txt", "r") as f: with open("workers/comfyui-json/misc/test_prompts.txt", "r") as f:
test_prompts = f.readlines() test_prompts = f.readlines()