Update src/facerender/modules/make_animation.py

src/facerender/modules/make_animation.py

@@ -148,11 +148,11 @@ def make_animation(source_image, source_semantics, target_semantics,
                    yaw_c_seq=None, pitch_c_seq=None, roll_c_seq=None,
                    use_exp=True):
 
-    # device='cuda'
+    device='cuda'
     # Move inputs to GPU
-    # source_image = source_image.to(device)
-    # source_semantics = source_semantics.to(device)
-    # target_semantics = target_semantics.to(device)
+    source_image = source_image.to(device)
+    source_semantics = source_semantics.to(device)
+    target_semantics = target_semantics.to(device)
 
     with torch.no_grad():  # No gradients needed
         predictions = []