From ba978bc0e20a2e66ddc2dc23c09b3a137734a56a Mon Sep 17 00:00:00 2001 From: Tara Ding <38710454+windtara0619@users.noreply.github.com> Date: Mon, 27 Apr 2026 22:13:55 -0700 Subject: [PATCH] Fix format --- main.py | 10 ++++------ 1 file changed, 4 insertions(+), 6 deletions(-) diff --git a/main.py b/main.py index 07c0dd359..8be3036ab 100644 --- a/main.py +++ b/main.py @@ -310,8 +310,7 @@ def prompt_worker(q, server_instance): extra_data[k] = sensitive[k] benchmark_mode = args.benchmark_server_only - if not benchmark_mode: - asset_seeder.pause() + asset_seeder.pause() e.execute(item[2], prompt_id, extra_data, item[4]) need_gc = True @@ -342,7 +341,7 @@ def prompt_worker(q, server_instance): "nodes": e.node_timing_ms, } - if not benchmark_mode and not asset_seeder.is_disabled(): + if not asset_seeder.is_disabled(): paths = _collect_output_absolute_paths(e.history_result) register_output_files(paths, job_id=prompt_id) @@ -368,10 +367,9 @@ def prompt_worker(q, server_instance): need_gc = False hook_breaker_ac10a0.restore_functions() - if not benchmark_mode and not asset_seeder.is_disabled(): + if not asset_seeder.is_disabled(): asset_seeder.enqueue_enrich(roots=("output",), compute_hashes=True) - if not benchmark_mode: - asset_seeder.resume() + asset_seeder.resume() async def run(server_instance, address='', port=8188, verbose=True, call_on_start=None):