From 9a8f4795d16551628c8bfaf8594d6a7237e33c4d Mon Sep 17 00:00:00 2001
From: bennykok <itechbenny@gmail.com>
Date: Mon, 9 Dec 2024 00:35:12 +0800
Subject: [PATCH] fix log

---
 perf_counter.py | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/perf_counter.py b/perf_counter.py
index 642bb7e..1a7f87d 100644
--- a/perf_counter.py
+++ b/perf_counter.py
@@ -8,6 +8,12 @@ from tabulate import tabulate
 from model_management import get_torch_device
 import psutil
 
+
+from logging import basicConfig, getLogger
+
+logger = getLogger("comfy-deploy")
+basicConfig(level="INFO")  # You can adjust the logging level as needed
+
 prompt_server = server.PromptServer.instance
 
 NODE_EXECUTION_TIMES = {}  # New dictionary to store node execution times
@@ -131,7 +137,7 @@ def swizzle_send_sync(self, event, data, sid=None):
             ])
             
             # print("\n=== Node Execution Times ===")
-            print(tabulate(table_data, headers=headers, tablefmt="grid"))
+            logger.info(tabulate(table_data, headers=headers, tablefmt="grid"))
             # print("========================\n")