Spaces:
Paused
Paused
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):
|