Skip to content

GitLab

"...data/git@developer.sourcefind.cn:wangsen/paddle_dbnet.git" did not exist on "4290c697f726f12cbb9d99497bd2a1488df95af7"
Commit 9bf67c4c authored by comfyanonymous's avatar comfyanonymous
Browse files

Print prompt execution time.

parent e7b9d2c0
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 0 deletions
execution.py
View file @ 9bf67c4c
...@@ -6,6 +6,7 @@ import threading ...@@ -6,6 +6,7 @@ import threading
import heapq import heapq
import traceback import traceback
import gc import gc
import time
import torch import torch
import nodes import nodes
...@@ -215,6 +216,7 @@ class PromptExecutor: ...@@ -215,6 +216,7 @@ class PromptExecutor:
else: else:
self.server.client_id = None self.server.client_id = None
execution_start_time = time.perf_counter()
if self.server.client_id is not None: if self.server.client_id is not None:
self.server.send_sync("execution_start", { "prompt_id": prompt_id}, self.server.client_id) self.server.send_sync("execution_start", { "prompt_id": prompt_id}, self.server.client_id)
...@@ -272,6 +274,7 @@ class PromptExecutor: ...@@ -272,6 +274,7 @@ class PromptExecutor:
if self.server.client_id is not None: if self.server.client_id is not None:
self.server.send_sync("executing", { "node": None, "prompt_id": prompt_id }, self.server.client_id) self.server.send_sync("executing", { "node": None, "prompt_id": prompt_id }, self.server.client_id)
print("Prompt executed in {:.2f} seconds".format(time.perf_counter() - execution_start_time))
gc.collect() gc.collect()
comfy.model_management.soft_empty_cache() comfy.model_management.soft_empty_cache()
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment