SadTalker

Runtime error

vinthony commited on Jun 20, 2023

Commit

99e1f07

•

1 Parent(s): 2eea0f8

Update src/face3d/extract_kp_videos.py

Files changed (1) hide show

src/face3d/extract_kp_videos.py CHANGED Viewed

@@ -13,7 +13,7 @@ from torch.multiprocessing import Pool, Process, set_start_method
 class KeypointExtractor():
     def __init__(self, device):
-        self.detector = face_alignment.FaceAlignment(face_alignment.LandmarksType._2D, device='cuda:0')
     def extract_keypoint(self, images, name=None, info=True):
         if isinstance(images, list):

 class KeypointExtractor():
     def __init__(self, device):
+        self.detector = face_alignment.FaceAlignment(face_alignment.LandmarksType._2D, device='cpu')
     def extract_keypoint(self, images, name=None, info=True):
         if isinstance(images, list):