Update gen_images.py

VinAIResearch · May 16, 2023 · d2bae20 · d2bae20
1 parent 41ba255
commit d2bae20
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/gen_images.py b/gen_images.py
@@ -205,7 +205,7 @@ def generate_images(
 
         # Construct an input coordinates and pass to the creps generator.
         if hasattr(G.synthesis.b4, "input"):
-            G.synthesis.input.coords = make_coords(G.img_resolution, scale).to(device)
+            G.synthesis.b4.input.coords = make_coords(G.img_resolution, scale).to(device)
 
         img = G(z, label, truncation_psi=truncation_psi, noise_mode=noise_mode)
         img = (img.permute(0, 2, 3, 1) * 127.5 + 128).clamp(0, 255).to(torch.uint8)