Update src/pipelines/pipeline_echo_mimic_pose_acc.py
Browse files
src/pipelines/pipeline_echo_mimic_pose_acc.py
CHANGED
|
@@ -87,7 +87,8 @@ class AudioPose2VideoPipeline(DiffusionPipeline):
|
|
| 87 |
else:
|
| 88 |
raise ImportError("Please install accelerate via `pip install accelerate`")
|
| 89 |
|
| 90 |
-
device = torch.device(f"cuda:{gpu_id}")
|
|
|
|
| 91 |
|
| 92 |
for cpu_offloaded_model in [self.unet, self.text_encoder, self.vae]:
|
| 93 |
if cpu_offloaded_model is not None:
|
|
|
|
| 87 |
else:
|
| 88 |
raise ImportError("Please install accelerate via `pip install accelerate`")
|
| 89 |
|
| 90 |
+
# device = torch.device(f"cuda:{gpu_id}")
|
| 91 |
+
device = torch.device(f"cpu:{gpu_id}")
|
| 92 |
|
| 93 |
for cpu_offloaded_model in [self.unet, self.text_encoder, self.vae]:
|
| 94 |
if cpu_offloaded_model is not None:
|