INFERENCE_STEPS,
        generator=generator,
        num_images_per_prompt=NUM_IMAGES_PER_PROMPT
    ).images
end = time.time_ns()

distilled_sd = f"{(end - start) / 1e6:.1f}"
print(f"Execution time -- {distilled_sd} ms\n")
"Execution time -- 29884.2 ms"  original Stable Diffusion (45781.5 ms)  distilled Stable Diffusion (29884.2 ms)  Tiny AutoEncoder To speed inference up even more, use a tiny distilled version of the Stable Diffusion VAE to 