diff --git a/apply_factor.py b/apply_factor.py index 9bf7f0ed..7dd22a2b 100755 --- a/apply_factor.py +++ b/apply_factor.py @@ -67,8 +67,8 @@ trunc = g.mean_latent(4096) - if torch_seed > 0: - torch.manual_seed(torch_seed) + if args.torch_seed > 0: + torch.manual_seed(args.torch_seed) latent = torch.randn(args.n_sample, 512, device=args.device) latent = g.get_latent(latent)