How to run profiler on txt2img?

[f0lie]

I looked at the codebase and there isn't an import of profiler. pytorch does have a profiler available though.

https://pytorch.org/tutorials/recipes/recipes/profiler_recipe.html

For example, we can profile the time taken with cuda.

model = models.resnet18().cuda()
inputs = torch.randn(5, 3, 224, 224).cuda()

with profile(activities=[
        ProfilerActivity.CPU, ProfilerActivity.CUDA], record_shapes=True) as prof:
    with record_function("model_inference"):
        model(inputs)

print(prof.key_averages().table(sort_by="cuda_time_total", row_limit=10))

stable-diffusion-webui/modules/processing.py

Line 737 in 0cc0ee1

class StableDiffusionProcessingTxt2Img(StableDiffusionProcessing):

I was interested in profiling hires to get better performance, but it is not exactly clear where to add this profiler to the code.