|
|
@ -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, |
|
|
|
}, { |
|
|
|
}, { |
|
|
|