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

Use count_workload() function for cost

Browse Source
This commit is contained in:
Rob Ballantyne
2025-08-25 18:31:10 +01:00
parent ba74ac8136
commit 16b414676e
1 changed files with 5 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
workers/comfyui-json/client.py
+5 -3
View File
@@ -9,6 +9,7 @@ import requests
from lib.test_utils import print_truncate_res from lib.test_utils import print_truncate_res
from utils.endpoint_util import Endpoint from utils.endpoint_util import Endpoint
from utils.ssl import get_cert_file_path from utils.ssl import get_cert_file_path
from .data_types import count_workload
logging.basicConfig( logging.basicConfig(
level=logging.DEBUG, level=logging.DEBUG,
@@ -58,7 +59,9 @@ def call_text2image_workflow(
return None return None
WORKER_ENDPOINT = "/generate/sync" WORKER_ENDPOINT = "/generate/sync"
COST = 100
# This worker has concurrency = 1. All workloads have cost value 1.0
COST = count_workload()
# Route to get worker URL # Route to get worker URL
route_payload = { route_payload = {
@@ -101,8 +104,7 @@ def call_text2image_workflow(
"seed": random.randint(0, 2**32 - 1) "seed": random.randint(0, 2**32 - 1)
}, },
"workflow_json": {} # Empty since using modifier approach "workflow_json": {} # Empty since using modifier approach
}, }
"expected_time": 30.0 # Expected 30 seconds on RTX4090
} }
req_data = dict(payload=worker_payload, auth_data=auth_data) req_data = dict(payload=worker_payload, auth_data=auth_data)