Update src/facerender/modules/make_animation.py

src/facerender/modules/make_animation.py

@@ -149,7 +149,10 @@ def make_animation(source_image, source_semantics, target_semantics,
                    use_exp=True):
 
     device='cuda'
-    # Move inputs to GPU
+    generator = torch.nn.DataParallel(generator).to(device)
+    kp_detector = torch.nn.DataParallel(kp_detector).to(device)
+    mapping = torch.nn.DataParallel(mapping).to(device)
+    
     source_image = source_image.to(device)
     source_semantics = source_semantics.to(device)
     target_semantics = target_semantics.to(device)