Update src/pipelines/pipeline_echo_mimic.py

src/pipelines/pipeline_echo_mimic.py

@@ -93,7 +93,8 @@ class Audio2VideoPipeline(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(f"cpu:{gpu_id}")
 
         for cpu_offloaded_model in [self.unet, self.text_encoder, self.vae]:
             if cpu_offloaded_model is not None: