Update videoretalking/third_part/GPEN/face_model/op/fused_act.py
Browse files
videoretalking/third_part/GPEN/face_model/op/fused_act.py
CHANGED
|
@@ -19,7 +19,7 @@ from torch.utils.cpp_extension import load, _import_module_from_library
|
|
| 19 |
# )
|
| 20 |
|
| 21 |
|
| 22 |
-
|
| 23 |
|
| 24 |
|
| 25 |
class FusedLeakyReLUFunctionBackward(Function):
|
|
|
|
| 19 |
# )
|
| 20 |
|
| 21 |
|
| 22 |
+
fused = _import_module_from_library('fused', '/tmp/torch_extensions/fused', True)
|
| 23 |
|
| 24 |
|
| 25 |
class FusedLeakyReLUFunctionBackward(Function):
|