Spaces:
Running
Running
Yannic Kilcher
commited on
Commit
•
99b07a7
1
Parent(s):
2934920
bugfix
Browse files- 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,
|
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 |
|