Hugging Face's logo

Spaces:
huggan
/
sefa
Runtime error

App Files Community
Johannes Kolbe commited on
Commit
b1da9de
•
1 Parent(s): 424816e

add more latent codes

Browse files
Files changed (4) hide show
  1. app.py +1 -1
  2. latent_codes/pggan_celebahq1024_latents.npy +0 -0
  3. latent_codes/stylegan_car512_latents.npy +0 -0
  4. latent_codes/stylegan_cat256_latents.npy +0 -0
app.py CHANGED
@@ -27,7 +27,7 @@ def factorize_model(model, layer_idx):
27
 
28
  def sample(model, gan_type, num=1):
29
  """Samples latent codes."""
30
- codes = torch.randn(num, model.z_space_dim).cuda()
31
  if gan_type == 'pggan':
32
  codes = model.layer0.pixel_norm(codes)
33
  elif gan_type == 'stylegan':
 
27
 
28
  def sample(model, gan_type, num=1):
29
  """Samples latent codes."""
30
+ codes = torch.randn(num, model.z_space_dim)
31
  if gan_type == 'pggan':
32
  codes = model.layer0.pixel_norm(codes)
33
  elif gan_type == 'stylegan':
latent_codes/pggan_celebahq1024_latents.npy ADDED
Binary file (20.6 kB). View file
 
latent_codes/stylegan_car512_latents.npy ADDED
Binary file (328 kB). View file
 
latent_codes/stylegan_cat256_latents.npy ADDED
Binary file (287 kB). View file