feng2022 commited on
Commit
c0972cf
1 Parent(s): 2e0063b

Update Time_TravelRephotography/op/fused_act.py

Browse files
Time_TravelRephotography/op/fused_act.py CHANGED
@@ -82,6 +82,6 @@ class FusedLeakyReLU(nn.Module):
82
  return fused_leaky_relu(input, self.bias, self.negative_slope, self.scale)
83
 
84
 
85
- def fused_leaky_relu(input, bias):
86
- return FusedLeakyReLU.apply(input, bias)
87
 
 
82
  return fused_leaky_relu(input, self.bias, self.negative_slope, self.scale)
83
 
84
 
85
+ def fused_leaky_relu(input):
86
+ return FusedLeakyReLU.apply(input)
87