Browse Source

Fix stable cascade img2img not working with all resolutions.

pull/3088/head
comfyanonymous 8 months ago
parent
commit
b1a16d4500
  1. 2
      comfy_extras/nodes_stable_cascade.py

2
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) s = comfy.utils.common_upscale(image.movedim(-1,1), out_width, out_height, "bicubic", "center").movedim(1,-1)
c_latent = vae.encode(s[:,:,:,:3]) 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 ({ return ({
"samples": c_latent, "samples": c_latent,
}, { }, {

Loading…
Cancel
Save