mirrors/Fooocus
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Merge pull request #464 from MoonRide303/log-execution-time

Browse Source 
Added information about diffusion and total execution time (#463)
This commit is contained in:
MoonRide303 2023-09-21 11:07:17 +02:00 committed by GitHub
commit e5ca884886
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 10 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
modules/async_worker.py
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),

5
webui.py
View File

@ -15,6 +15,8 @@ from modules.sdxl_styles import style_keys, aspect_ratios, fooocus_expansion, de
def generate_clicked(*args):
execution_start_time = time.perf_counter()
yield gr.update(visible=True, value=modules.html.make_progress_html(1, 'Initializing ...')), \
gr.update(visible=True, value=None), \
gr.update(visible=False)
@ -36,6 +38,9 @@ def generate_clicked(*args):
gr.update(visible=False), \
gr.update(visible=True, value=product)
finished = True
execution_time = time.perf_counter() - execution_start_time
print(f'Total time: {execution_time:.2f} seconds')
return