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

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

@@ -19,7 +19,7 @@ from torch.utils.cpp_extension import load, _import_module_from_library
 #     )
 
 
-fused = _import_module_from_library('fused', '/tmp/torch_extensions/fused', True)
+# fused = _import_module_from_library('fused', '/tmp/torch_extensions/fused', True)
 
 
 class FusedLeakyReLUFunctionBackward(Function):
@@ -90,7 +90,7 @@ class FusedLeakyReLU(nn.Module):
 
 
 def fused_leaky_relu(input, bias, negative_slope=0.2, scale=2 ** 0.5, device='cpu'):
-    if platform.system() == 'Linux' and torch.cuda.is_available() and device != 'cpu':
-        return FusedLeakyReLUFunction.apply(input, bias, negative_slope, scale)
-    else:
-        return scale * F.leaky_relu(input + bias.view((1, -1)+(1,)*(len(input.shape)-2)), negative_slope=negative_slope)
+    # if platform.system() == 'Linux' and torch.cuda.is_available() and device != 'cpu':
+    #     return FusedLeakyReLUFunction.apply(input, bias, negative_slope, scale)
+    # else:
+    return scale * F.leaky_relu(input + bias.view((1, -1)+(1,)*(len(input.shape)-2)), negative_slope=negative_slope)