Update Time_TravelRephotography/model.py

Time_TravelRephotography/model.py

@@ -152,7 +152,7 @@ class EqualLinear(nn.Module):
     def forward(self, input):
         if self.activation:
             out = F.linear(input, self.weight * self.scale)
-            out = fused_leaky_relu(out, self.bias * self.lr_mul)
+            out = FusedLeakyReLU(out, self.bias * self.lr_mul)
 
         else:
             out = F.linear(