Public/automatic
1
0
Fork 0
mirror of https://github.com/vladmandic/automatic synced 2026-09-19 01:04:32 +02:00
Code Issues Packages Projects Releases Wiki Activity

add profiler

Browse Source
This commit is contained in:
Vladimir Mandic
2023-02-28 10:04:25 -05:00
parent 5e4a9337a1
commit 679320a764
5 changed files with 24 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/processing.py
+8
View File
@@ -483,6 +483,14 @@ def process_images(p: StableDiffusionProcessing) -> Processed:
if k == 'sd_vae':
sd_vae.reload_vae_weights()
"""
import torch.profiler
with torch.profiler.profile(activities=[torch.profiler.ProfilerActivity.CPU, torch.profiler.ProfilerActivity.CUDA], record_shapes=True, with_modules=True) as prof:
with torch.profiler.record_function("process_images"):
res = process_images_inner(p)
print(prof.key_averages().table(sort_by="cuda_time_total", row_limit=15))
"""
res = process_images_inner(p)
finally: