Fix typo

rosinality · Nov 6, 2023 · 37d7a6b · 37d7a6b
1 parent 788d9b4
commit 37d7a6b
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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)