refactor: simplify unpacking of latent samples shape in PadAndResize

2025-09-28 20:50:19 +02:00
parent bece36e573
commit d541486f4e
1 changed files with 1 additions and 1 deletions
@@ -41,7 +41,7 @@ class PadAndResize:
    
    def resize(self, image, latent, mode):
        b, c, sh, sw = image.shape
-        _, _, th, tw = latent['samples'].shape
+        *_, th, tw = latent['samples'].shape
        th, tw = th * 8, tw * 8
        if (th, tw) == (sh, sw):
            return image