Update rerender_a_video.py

Author: chaowenguo0

examples/community/rerender_a_video.py

@@ -1104,6 +1104,9 @@ def denoising_loop(latents, mask=None, xtrg=None, noise_rescale=None):
                             progress_bar.update()
                             if callback is not None and i % callback_steps == 0:
                                 callback(i, t, latents)
+                                
+                        if XLA_AVAILABLE:
+                            xm.mark_step()
 
                     return latents