Spaces:
Runtime error
Runtime error
Update Time_TravelRephotography/op/fused_act.py
Browse files
Time_TravelRephotography/op/fused_act.py
CHANGED
@@ -83,5 +83,5 @@ class FusedLeakyReLU(nn.Module):
|
|
83 |
|
84 |
|
85 |
def fused_leaky_relu(input, bias, negative_slope=0.2, scale=2 ** 0.5):
|
86 |
-
return
|
87 |
|
|
|
83 |
|
84 |
|
85 |
def fused_leaky_relu(input, bias, negative_slope=0.2, scale=2 ** 0.5):
|
86 |
+
return FusedLeakyReLU.apply(input, bias, negative_slope, scale)
|
87 |
|