up your inference code even more, use torch_compile. You should wrap torch.compile around the most intensive component in the pipeline which is typically the UNet:   Copied pipeline.unet = torch.compile(pipeline.unet, mode="reduce-overhead", fullgraph=True) Learn more in the Reduce memory usage and Torch 2.0 guides.
