Added information about diffusion and total execution time (#463)

This commit is contained in:
MoonRide303
2023-09-21 10:54:09 +02:00
parent 7e031b1b03
commit 726f684ff2
2 changed files with 10 additions and 0 deletions
+5
View File
@@ -336,6 +336,8 @@ def worker():
outputs.append(['preview', (13, 'Starting tasks ...', None)])
for current_task_id, task in enumerate(tasks):
try:
execution_start_time = time.perf_counter()
imgs = pipeline.process_diffusion(
positive_cond=task['c'],
negative_cond=task['uc'],
@@ -353,6 +355,9 @@ def worker():
if inpaint_worker.current_task is not None:
imgs = [inpaint_worker.current_task.post_process(x) for x in imgs]
execution_time = time.perf_counter() - execution_start_time
print(f'Diffusion time: {execution_time:.2f} seconds')
for x in imgs:
d = [
('Prompt', raw_prompt),