From c860cc612bc9c9a7377dc28b140bb41219e20d6f Mon Sep 17 00:00:00 2001 From: Richard Yu Date: Thu, 4 Dec 2025 12:52:36 -0800 Subject: [PATCH] revert main.py fully --- main.py | 9 ++++----- 1 file changed, 4 insertions(+), 5 deletions(-) diff --git a/main.py b/main.py index 1eb4423b8..0cd815d9e 100644 --- a/main.py +++ b/main.py @@ -229,20 +229,19 @@ def prompt_worker(q, server_instance): e.execute(item[2], prompt_id, extra_data, item[4]) need_gc = True - current_time = time.perf_counter() - execution_time = current_time - execution_start_time - remove_sensitive = lambda prompt: prompt[:5] + prompt[6:] q.task_done(item_id, e.history_result, status=execution.PromptQueue.ExecutionStatus( status_str='success' if e.success else 'error', completed=e.success, - messages=e.status_messages), - process_item=remove_sensitive) + messages=e.status_messages), process_item=remove_sensitive) if server_instance.client_id is not None: server_instance.send_sync("executing", {"node": None, "prompt_id": prompt_id}, server_instance.client_id) + current_time = time.perf_counter() + execution_time = current_time - execution_start_time + # Log Time in a more readable way after 10 minutes if execution_time > 600: execution_time = time.strftime("%H:%M:%S", time.gmtime(execution_time))