mirror of
https://github.com/comfyanonymous/ComfyUI.git
synced 2026-01-10 06:10:50 +08:00
23 lines
492 B
Python
23 lines
492 B
Python
from comfy.node_helpers import export_custom_nodes
|
|
from comfy.nodes.package_typing import CustomNode, InputTypes
|
|
|
|
|
|
class OutputTensor(CustomNode):
|
|
@classmethod
|
|
def INPUT_TYPES(cls) -> InputTypes:
|
|
return {
|
|
"required": {
|
|
"tensor": ("IMAGE,AUDIO,VIDEO", {})
|
|
}
|
|
}
|
|
|
|
RETURN_TYPES = ()
|
|
OUTPUT_NODE = True
|
|
FUNCTION = "execute"
|
|
|
|
def execute(self, tensor):
|
|
return {"ui": {"tensor": tensor}}
|
|
|
|
|
|
export_custom_nodes()
|