feng2022 commited on
Commit
3644a67
1 Parent(s): 70ebff6

Update Time_TravelRephotography/model.py

Browse files
Files changed (1) hide show
  1. Time_TravelRephotography/model.py +2 -1
Time_TravelRephotography/model.py CHANGED
@@ -10,7 +10,8 @@ from torch.nn import functional as F
10
  from torch.autograd import Function
11
 
12
  #from op import FusedLeakyReLU, fused_leaky_relu, upfirdn2d
13
- from torch_utils.ops import bias_act, upfirdn2d
 
14
 
15
 
16
  class PixelNorm(nn.Module):
 
10
  from torch.autograd import Function
11
 
12
  #from op import FusedLeakyReLU, fused_leaky_relu, upfirdn2d
13
+ from torch_utils.ops.bias_act import *
14
+ from upfirdn2d import *
15
 
16
 
17
  class PixelNorm(nn.Module):