fffiloni commited on
Commit
2a07f50
1 Parent(s): 7eb60e4

Update drag_gan.py

Browse files
Files changed (1) hide show
  1. drag_gan.py +1 -1
drag_gan.py CHANGED
@@ -118,7 +118,7 @@ class CustomGenerator(Generator):
118
  def stylegan2(
119
  size=512,
120
  channel_multiplier=2,
121
- latent=256,
122
  n_mlp=8,
123
  ckpt='stylegan2-ffhq-config-f.pt'
124
  ):
 
118
  def stylegan2(
119
  size=512,
120
  channel_multiplier=2,
121
+ latent=512,
122
  n_mlp=8,
123
  ckpt='stylegan2-ffhq-config-f.pt'
124
  ):