mirror of
https://github.com/comfyanonymous/ComfyUI.git
synced 2026-06-26 09:49:26 +08:00
34 lines
889 B
Python
34 lines
889 B
Python
import sys
|
|
from typing_extensions import override
|
|
|
|
from comfy_api.latest import ComfyExtension, io
|
|
|
|
|
|
class SeedNode(io.ComfyNode):
|
|
@classmethod
|
|
def define_schema(cls):
|
|
return io.Schema(
|
|
node_id="SeedNode",
|
|
display_name="Seed",
|
|
search_aliases=["seed", "random"],
|
|
category="utilities",
|
|
inputs=[
|
|
io.Int.Input("seed", min=0, max=sys.maxsize, control_after_generate=io.ControlAfterGenerate.fixed),
|
|
],
|
|
outputs=[io.Int.Output(display_name="seed")],
|
|
)
|
|
|
|
@classmethod
|
|
def execute(cls, seed: int) -> io.NodeOutput:
|
|
return io.NodeOutput(seed)
|
|
|
|
|
|
class SeedExtension(ComfyExtension):
|
|
@override
|
|
async def get_node_list(self) -> list[type[io.ComfyNode]]:
|
|
return [SeedNode]
|
|
|
|
|
|
async def comfy_entrypoint() -> SeedExtension:
|
|
return SeedExtension()
|