Update node.py

2024-05-14 22:25:30 +09:00
parent cf9d07c084
commit 6161d0cbe1
1 changed files with 2 additions and 2 deletions
@@ -170,8 +170,8 @@ class SaveLatentBatch:
            if extension == "npy":
                np.save(path, latent.float().cpu().numpy())
            else:
-                original_size = (latent.shape[2] * 8, latent.shape[3] * 8)
-                crop_ltrb = (0, 0)
+                original_size = (latent.shape[1] * 8, latent.shape[2] * 8)
+                crop_ltrb = (0, 0, 0, 0)
                np.savez(
                    path, 
                    latents=latent.float().cpu().numpy(),