a

demo_dagan.py

@@ -55,7 +55,7 @@ def find_best_frame(source, driving, cpu=False):
         return kp
 
     fa = face_alignment.FaceAlignment(face_alignment.LandmarksType._2D, flip_input=True,
-                                      device='cpu' if cpu else 'cuda')
+                                      device='cpu')
     kp_source = fa.get_landmarks(255 * source)[0]
     kp_source = normalize_kp(kp_source)
     norm  = float('inf')

requirements.txt

@@ -28,8 +28,8 @@ sklearn==0.0
 tensorboard==1.15.0
 threadpoolctl==3.1.0
 tifffile==2021.11.2
-torch==1.10.1+cu111
-torchvision==0.11.2+cu111
+torch
+torchvision
 tqdm==4.64.0
 typing_extensions==4.2.0
 Werkzeug==2.1.1