Update videoretalking/third_part/GPEN/face_model/op/upfirdn2d.py

videoretalking/third_part/GPEN/face_model/op/upfirdn2d.py

@@ -147,12 +147,12 @@ class UpFirDn2d(Function):
 
 
 def upfirdn2d(input, kernel, up=1, down=1, pad=(0, 0), device='cpu'):
-    if platform.system() == 'Linux' and torch.cuda.is_available() and device != 'cpu':
-        out = UpFirDn2d.apply(
-            input, kernel, (up, up), (down, down), (pad[0], pad[1], pad[0], pad[1])
-        )
-    else:
-        out = upfirdn2d_native(input, kernel, up, up, down, down, pad[0], pad[1], pad[0], pad[1])
+    # if platform.system() == 'Linux' and torch.cuda.is_available() and device != 'cpu':
+    #     out = UpFirDn2d.apply(
+    #         input, kernel, (up, up), (down, down), (pad[0], pad[1], pad[0], pad[1])
+    #     )
+    # else:
+    out = upfirdn2d_native(input, kernel, up, up, down, down, pad[0], pad[1], pad[0], pad[1])
 
     return out