diff --git a/workers/ace/worker.py b/workers/ace/worker.py index deb5ef0..11524a6 100644 --- a/workers/ace/worker.py +++ b/workers/ace/worker.py @@ -38,10 +38,6 @@ benchmark_lyrics = [ "[verse]\nCrowds ignite in vibrant flare\nBeats explode through smoky air\nDJ robes replaced with flame\nPope on decks in holy frame\n\n[verse]\nLeather gleams in blinding light\nTurntables spin with sacred might\nChoirs echo in the bass\nHeaven pulses through the place\n\n[chorus]\nLift the roof and shake the floor\nSacred rhythm evermore\nLet the music take control\nFeel the blessing in your soul", ] - -def parse_request(json_msg): - return {"input" : json_msg} - benchmark_dataset = [ { "input": { @@ -171,11 +167,11 @@ worker_config = WorkerConfig( route="/generate/sync", allow_parallel_requests=False, max_queue_time=10.0, - request_parser=parse_request, benchmark_config=BenchmarkConfig( dataset=benchmark_dataset, runs=1 - ) + ), + workload_calculator= lambda _ : 1000.0 ) ], log_action_config=LogActionConfig( diff --git a/workers/wan/worker.py b/workers/wan/worker.py index 6d81469..174b5f4 100644 --- a/workers/wan/worker.py +++ b/workers/wan/worker.py @@ -38,9 +38,6 @@ benchmark_prompts = [ "Pope Francis DJ in leather jacket, mixing on giant console; dramatic.", ] -def parse_request(json_msg): - return {"input" : json_msg} - benchmark_dataset = [ { "input": { @@ -274,11 +271,11 @@ worker_config = WorkerConfig( route="/generate/sync", allow_parallel_requests=False, max_queue_time=10.0, - request_parser=parse_request, benchmark_config=BenchmarkConfig( dataset=benchmark_dataset, runs=1 - ) + ), + workload_calculator= lambda _ : 10000.0 ) ], log_action_config=LogActionConfig(