Added NIE on xla device

xora/pipelines/pipeline_xora_video.py

@@ -1010,7 +1010,10 @@ class XoraVideoPipeline(DiffusionPipeline):
                     current_timestep = current_timestep * (1 - conditioning_mask)
                 # Choose the appropriate context manager based on `mixed_precision`
                 if mixed_precision:
-                    context_manager = torch.autocast("cuda" if torch.cuda.is_available() else 'cpu', dtype=torch.bfloat16)
+                    if 'xla' in device.type:
+                        raise NotImplementedError("Mixed precision is not supported yet on XLA devices.")
+                        
+                    context_manager = torch.autocast(device, dtype=torch.bfloat16)
                 else:
                     context_manager = nullcontext()  # Dummy context manager