Update Time_TravelRephotography/op/fused_act.py

Time_TravelRephotography/op/fused_act.py

@@ -71,9 +71,10 @@ class FusedLeakyReLUFunction(Function):
 
 
 class FusedLeakyReLU(nn.Module):
-    def __init__(self, negative_slope=0.2, scale=2 ** 0.5):
+    def __init__(self, bias, negative_slope=0.2, scale=2 ** 0.5):
         super().__init__()
 
+        self.bias= bias
         self.negative_slope = negative_slope
         self.scale = scale