mirror of
https://github.com/comfyanonymous/ComfyUI.git
synced 2025-07-27 08:16:44 +00:00
39 lines
1.2 KiB
Python
39 lines
1.2 KiB
Python
from __future__ import annotations
|
|
|
|
import torch
|
|
|
|
import comfy.model_management
|
|
import nodes
|
|
from comfy_api.latest import io
|
|
|
|
|
|
class EmptyMochiLatentVideo(io.ComfyNode):
|
|
@classmethod
|
|
def define_schema(cls):
|
|
return io.Schema(
|
|
node_id="EmptyMochiLatentVideo_V3",
|
|
category="latent/video",
|
|
inputs=[
|
|
io.Int.Input("width", default=848, min=16, max=nodes.MAX_RESOLUTION, step=16),
|
|
io.Int.Input("height", default=480, min=16, max=nodes.MAX_RESOLUTION, step=16),
|
|
io.Int.Input("length", default=25, min=7, max=nodes.MAX_RESOLUTION, step=6),
|
|
io.Int.Input("batch_size", default=1, min=1, max=4096),
|
|
],
|
|
outputs=[
|
|
io.Latent.Output(),
|
|
],
|
|
)
|
|
|
|
@classmethod
|
|
def execute(cls, width, height, length, batch_size=1):
|
|
latent = torch.zeros(
|
|
[batch_size, 12, ((length - 1) // 6) + 1, height // 8, width // 8],
|
|
device=comfy.model_management.intermediate_device(),
|
|
)
|
|
return io.NodeOutput({"samples": latent})
|
|
|
|
|
|
NODES_LIST: list[type[io.ComfyNode]] = [
|
|
EmptyMochiLatentVideo,
|
|
]
|