Update rerender_a_video.py

Author: chaowenguo0

examples/community/rerender_a_video.py

@@ -30,12 +30,17 @@
 from diffusers.pipelines.controlnet.pipeline_controlnet_img2img import StableDiffusionControlNetImg2ImgPipeline
 from diffusers.pipelines.stable_diffusion.safety_checker import StableDiffusionSafetyChecker
 from diffusers.schedulers import KarrasDiffusionSchedulers
-from diffusers.utils import BaseOutput, deprecate, logging
+from diffusers.utils import is_torch_xla_available, BaseOutput, deprecate, logging
 from diffusers.utils.torch_utils import is_compiled_module, randn_tensor
 
+if is_torch_xla_available():
+    import torch_xla.core.xla_model as xm
 
-logger = logging.get_logger(__name__)  # pylint: disable=invalid-name
+    XLA_AVAILABLE = True
+else:
+    XLA_AVAILABLE = False
 
+logger = logging.get_logger(__name__)  # pylint: disable=invalid-name
 
 def coords_grid(b, h, w, homogeneous=False, device=None):
     y, x = torch.meshgrid(torch.arange(h), torch.arange(w))  # [H, W]