Spaces:

vinthony
/

SadTalker

Running on A10G

vinthony commited on Jun 20, 2023

Commit

2eea0f8

•

1 Parent(s): a12ed30

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='cuda:0')
     def extract_keypoint(self, images, name=None, info=True):
         if isinstance(images, list):