hr16 commited on
Commit
17dfcbb
·
1 Parent(s): 0aedaac

Hope that I can put 128-bit unsigned int seeds

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -38,14 +38,14 @@ def generate_images(G, args):
38
  if len(args.gpu) <= 1:
39
  noise_tensors = [[] for _ in noise_reference]
40
  for seed in seeds:
41
- rnd = np.random.RandomState(seed)
42
- latents.append(torch.from_numpy(rnd.randn(latent_size)))
43
  if len(args.gpu) <= 1:
44
  for i, ref in enumerate(noise_reference):
45
  noise_tensors[i].append(
46
- torch.from_numpy(rnd.randn(*ref.size()[1:])))
47
  if label_size:
48
- labels.append(torch.tensor([rnd.randint(0, label_size)]))
49
  latents = torch.stack(latents, dim=0).to(
50
  device=device, dtype=torch.float32)
51
  if labels:
 
38
  if len(args.gpu) <= 1:
39
  noise_tensors = [[] for _ in noise_reference]
40
  for seed in seeds:
41
+ rnd = np.random.default_rng(seed)
42
+ latents.append(torch.from_numpy(rnd.standard_normal(latent_size)))
43
  if len(args.gpu) <= 1:
44
  for i, ref in enumerate(noise_reference):
45
  noise_tensors[i].append(
46
+ torch.from_numpy(rnd.standard_normal(*ref.size()[1:])))
47
  if label_size:
48
+ labels.append(torch.tensor([rnd.integers(0, label_size)]))
49
  latents = torch.stack(latents, dim=0).to(
50
  device=device, dtype=torch.float32)
51
  if labels: