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

fix: ordered dict

Browse Source
This commit is contained in:
bennykok 2024-12-09 18:13:36 +08:00
parent 40ec37e58f
commit f39d216326
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
custom_routes.py
View File

@ -29,6 +29,7 @@ import atexit
from model_management import get_torch_device from model_management import get_torch_device
import torch import torch
import psutil import psutil
from collections import OrderedDict
# Global session # Global session
client_session = None client_session = None
@ -1124,7 +1125,7 @@ async def proxy_to_comfydeploy(request):
prompt_server = server.PromptServer.instance prompt_server = server.PromptServer.instance
NODE_EXECUTION_TIMES = {} # New dictionary to store node execution times NODE_EXECUTION_TIMES = OrderedDict() # New dictionary to store node execution times
CURRENT_START_EXECUTION_DATA = None CURRENT_START_EXECUTION_DATA = None
def get_peak_memory(): def get_peak_memory():
@ -1230,7 +1231,7 @@ def swizzle_send_sync(self, event, data, sid=None):
global CURRENT_START_EXECUTION_DATA global CURRENT_START_EXECUTION_DATA
if event == "execution_start": if event == "execution_start":
global NODE_EXECUTION_TIMES global NODE_EXECUTION_TIMES
NODE_EXECUTION_TIMES = {} # Reset execution times at start NODE_EXECUTION_TIMES = OrderedDict() # Reset execution times at start
CURRENT_START_EXECUTION_DATA = dict( CURRENT_START_EXECUTION_DATA = dict(
start_perf_time=time.perf_counter(), start_perf_time=time.perf_counter(),
nodes_start_perf_time={}, nodes_start_perf_time={},