add

utils/other_tools_hf.py

@@ -694,11 +694,12 @@ def render_one_sequence_no_gt(
     import moviepy.editor as mp
     import librosa
     
+    device = "cpu"
     model = smplx.create(model_folder, model_type=model_type,
                          gender=gender, use_face_contour=use_face_contour,
                          num_betas=num_betas,
                          num_expression_coeffs=num_expression_coeffs,
-                         ext=ext, use_pca=False).cuda()
+                         ext=ext, use_pca=False).to(device)
     
     #data_npz = np.load(f"{output_dir}{res_npz_path}.npz")
     data_np_body = np.load(res_npz_path, allow_pickle=True)
@@ -715,12 +716,12 @@ def render_one_sequence_no_gt(
     seconds = 1
     #data_npz["jaw_pose"].shape[0]
     n = data_np_body["poses"].shape[0]
-    beta = torch.from_numpy(data_np_body["betas"]).to(torch.float32).unsqueeze(0).cuda()
+    beta = torch.from_numpy(data_np_body["betas"]).to(torch.float32).unsqueeze(0).to(device)
     beta = beta.repeat(n, 1)
-    expression = torch.from_numpy(data_np_body["expressions"][:n]).to(torch.float32).cuda()
-    jaw_pose = torch.from_numpy(data_np_body["poses"][:n, 66:69]).to(torch.float32).cuda()
-    pose = torch.from_numpy(data_np_body["poses"][:n]).to(torch.float32).cuda()
-    transl = torch.from_numpy(data_np_body["trans"][:n]).to(torch.float32).cuda()
+    expression = torch.from_numpy(data_np_body["expressions"][:n]).to(torch.float32).to(device)
+    jaw_pose = torch.from_numpy(data_np_body["poses"][:n, 66:69]).to(torch.float32).to(device)
+    pose = torch.from_numpy(data_np_body["poses"][:n]).to(torch.float32).to(device)
+    transl = torch.from_numpy(data_np_body["trans"][:n]).to(torch.float32).to(device)
     # print(beta.shape, expression.shape, jaw_pose.shape, pose.shape, transl.shape, pose[:,:3].shape)
     output = model(betas=beta, transl=transl, expression=expression, jaw_pose=jaw_pose,
         global_orient=pose[:,:3], body_pose=pose[:,3:21*3+3], left_hand_pose=pose[:,25*3:40*3], right_hand_pose=pose[:,40*3:55*3],