Arnaudding001 commited on
Commit
fddc503
·
1 Parent(s): e1426f2

Update stylegan_model.py

Browse files
Files changed (1) hide show
  1. stylegan_model.py +1 -2
stylegan_model.py CHANGED
@@ -8,8 +8,7 @@ from torch import nn
8
  from torch.nn import functional as F
9
  from torch.autograd import Function
10
 
11
- import FusedLeakyReLU
12
- import fused_leaky_relu
13
  import stylegan_op_upfirdn2d
14
  import stylegan_op_conv2d_gradfix
15
 
 
8
  from torch.nn import functional as F
9
  from torch.autograd import Function
10
 
11
+ from stylegan_op_fused_act import FusedLeakyReLU, fused_leaky_relu
 
12
  import stylegan_op_upfirdn2d
13
  import stylegan_op_conv2d_gradfix
14