Skip to content

Commit 20f5e47

Browse files
Use LatentCutToBatch instead. (Comfy-Org#13815)
Removed VAEDecodeVideoFramewise from nodes_wandancer.py.
1 parent 3200f28 commit 20f5e47

1 file changed

Lines changed: 0 additions & 31 deletions

File tree

comfy_extras/nodes_wandancer.py

Lines changed: 0 additions & 31 deletions
Original file line numberDiff line numberDiff line change
@@ -842,36 +842,6 @@ def execute(cls, positive, negative, vae, width, height, length, start_image=Non
842842
return io.NodeOutput(positive, negative, out_latent)
843843

844844

845-
class VAEDecodeVideoFramewise(io.ComfyNode):
846-
@classmethod
847-
def define_schema(cls):
848-
return io.Schema(
849-
node_id="VAEDecodeVideoFramewise",
850-
category="latent",
851-
description="Decodes video latents one latent at a time.",
852-
search_aliases=["decode", "decode latent", "latent to image", "render latent"],
853-
inputs=[
854-
io.Latent.Input("samples", tooltip="The latent to be decoded."),
855-
io.Vae.Input("vae", tooltip="The VAE model used for decoding the latent."),
856-
],
857-
outputs=[
858-
io.Image.Output(tooltip="The decoded images."),
859-
],
860-
)
861-
862-
@classmethod
863-
def execute(cls, vae, samples) -> io.NodeOutput:
864-
latent = samples["samples"]
865-
if latent.is_nested:
866-
latent = latent.unbind()[0]
867-
868-
# reshape temporal dimension into batch
869-
B, C, T, H, W = latent.shape
870-
latent_batched = latent.transpose(1, 2).reshape(B * T, C, 1, H, W)
871-
images = vae.decode(latent_batched).squeeze(1)
872-
873-
return io.NodeOutput(images)
874-
875845
class WanDancerPadKeyframes(io.ComfyNode):
876846
@classmethod
877847
def define_schema(cls):
@@ -992,7 +962,6 @@ class WanDancerExtension(ComfyExtension):
992962
async def get_node_list(self) -> list[type[io.ComfyNode]]:
993963
return [
994964
WanDancerVideo,
995-
VAEDecodeVideoFramewise,
996965
WanDancerEncodeAudio,
997966
WanDancerPadKeyframes,
998967
WanDancerPadKeyframesList,

0 commit comments

Comments
 (0)