Update utils/digital_human/realtime_inference.py

utils/digital_human/realtime_inference.py

@@ -111,8 +111,8 @@ def load_pose_model(model_dir):
     )
 
     config_file = r"./utils/digital_human/musetalk/utils/dwpose/rtmpose-l_8xb32-270e_coco-ubody-wholebody-384x288.py"
-    pose_model = init_model(config_file, dw_pose_path, device="cuda")
-
+    pose_model = init_model(config_file, dw_pose_path, device="cpu")
+    #pose_model = init_model(config_file, dw_pose_path, device="cuda")
     return pose_model
 
 
@@ -397,7 +397,8 @@ class Avatar:
             audio_feature_batch = self.model_handler.pe(audio_feature_batch)
             latent_batch = latent_batch.to(dtype=self.model_handler.unet.model.dtype)
 
-            timesteps = torch.tensor([0], device="cuda")
+#            timesteps = torch.tensor([0], device="cuda")
+            timesteps = torch.tensor([0], device="cpu")
             pred_latents = self.model_handler.unet.model(
                 latent_batch, timesteps, encoder_hidden_states=audio_feature_batch
             ).sample