Fix stable cascade img2img not working with all resolutions.

11 months ago · b1a16d4500
parent cacb022c4a
commit b1a16d4500
1 changed files with 1 additions and 1 deletions
--- a/comfy_extras/nodes_stable_cascade.py
+++ b/comfy_extras/nodes_stable_cascade.py
@ -74,7 +74,7 @@ class StableCascade_StageC_VAEEncode:
        s = comfy.utils.common_upscale(image.movedim(-1,1), out_width, out_height, "bicubic", "center").movedim(1,-1)

        c_latent = vae.encode(s[:,:,:,:3])
-        b_latent = torch.zeros([c_latent.shape[0], 4, height // 4, width // 4])
+        b_latent = torch.zeros([c_latent.shape[0], 4, (height // 8) * 2, (width // 8) * 2])
        return ({
            "samples": c_latent,
        }, {