From d6c030fd379ac172f8d7b8d8f9da4cdeb8c2271c Mon Sep 17 00:00:00 2001 From: chaowenguo Date: Fri, 10 Jan 2025 13:03:41 -0800 Subject: [PATCH] add the xm.mark_step for the first denosing loop (#10530) * Update rerender_a_video.py * Update rerender_a_video.py * Update examples/community/rerender_a_video.py Co-authored-by: hlky * Update rerender_a_video.py * make style --------- Co-authored-by: hlky Co-authored-by: YiYi Xu --- examples/community/rerender_a_video.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/examples/community/rerender_a_video.py b/examples/community/rerender_a_video.py index a2830d8b0e..7e66bff51d 100644 --- a/examples/community/rerender_a_video.py +++ b/examples/community/rerender_a_video.py @@ -908,6 +908,9 @@ class RerenderAVideoPipeline(StableDiffusionControlNetImg2ImgPipeline): if callback is not None and i % callback_steps == 0: callback(i, t, latents) + if XLA_AVAILABLE: + xm.mark_step() + if not output_type == "latent": image = self.vae.decode(latents / self.vae.config.scaling_factor, return_dict=False)[0] else: