Spaces:

vinthony
/

SadTalker

Running on A10G

vinthony commited on Apr 9, 2023

Commit

684bb8e

•

1 Parent(s): 4142979

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=device)
     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):