update

utils/pipeline_magictime.py

@@ -27,6 +27,13 @@ from .unet import UNet3DConditionModel
 
 logger = logging.get_logger(__name__)  # pylint: disable=invalid-name
 
+if torch.cuda.is_available():
+    device = "cuda"
+elif torch.backends.mps.is_available():
+    device = "mps"
+else:
+    device = "cpu"
+
 @dataclass
 class MagicTimePipelineOutput(BaseOutput):
     videos: Union[torch.Tensor, np.ndarray]
@@ -120,7 +127,8 @@ class MagicTimePipeline(DiffusionPipeline):
         else:
             raise ImportError("Please install accelerate via `pip install accelerate`")
 
-        device = torch.device(f"cuda:{gpu_id}")
+        #device = torch.device(f"cuda:{gpu_id}")
+        device = torch.device(device)
 
         for cpu_offloaded_model in [self.unet, self.text_encoder, self.vae]:
             if cpu_offloaded_model is not None: