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

Update stylegan_model.py

Browse files
Files changed (1) hide show
  1. stylegan_model.py +4 -1
stylegan_model.py CHANGED
@@ -8,7 +8,10 @@ from torch import nn
8
  from torch.nn import functional as F
9
  from torch.autograd import Function
10
 
11
- from model.stylegan.op import FusedLeakyReLU, fused_leaky_relu, upfirdn2d, conv2d_gradfix
 
 
 
12
 
13
  class PixelNorm(nn.Module):
14
  def __init__(self):
 
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
 
16
  class PixelNorm(nn.Module):
17
  def __init__(self):