Yannic Kilcher commited on
Commit
f29d636
1 Parent(s): 3c0d676
Files changed (1) hide show
  1. interface.py +1 -1
interface.py CHANGED
@@ -44,7 +44,7 @@ def generate(num_images, interpolate):
44
  zs = torch.randn([num_images, G.z_dim])# latent codes
45
  with torch.no_grad():
46
  zs = zs.to(device)
47
- img = G(zs, None, force_fp32=True, truncation_psi=1, noise_mode='const')
48
  img = (img.permute(0, 2, 3, 1) * 127.5 + 128).clamp(0, 255).to(torch.uint8)
49
  return img.cpu().numpy()
50
 
 
44
  zs = torch.randn([num_images, G.z_dim])# latent codes
45
  with torch.no_grad():
46
  zs = zs.to(device)
47
+ img = G(zs, None, force_fp32=True, noise_mode='const')
48
  img = (img.permute(0, 2, 3, 1) * 127.5 + 128).clamp(0, 255).to(torch.uint8)
49
  return img.cpu().numpy()
50