chunzhi/comfyui-deploy
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix log

Browse Source
This commit is contained in:
bennykok 2024-12-09 00:35:12 +08:00
parent c0c617c5d2
commit 9a8f4795d1
1 changed files with 7 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
perf_counter.py
View File

@ -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")