mirror of
https://git.datalinker.icu/comfyanonymous/ComfyUI
synced 2026-09-09 09:17:03 +08:00
80 lines
2.8 KiB
Python
80 lines
2.8 KiB
Python
import json
|
|
from comfy_extras.nodes_custom_sampler import Noise_RandomNoise
|
|
|
|
|
|
class MD_VideoInputs:
|
|
"""One node to load all input parameters for video generation"""
|
|
|
|
@classmethod
|
|
def INPUT_TYPES(cls):
|
|
return {
|
|
"required": {
|
|
"image_url": (
|
|
"STRING",
|
|
),
|
|
"length": ("INT", {
|
|
"default": 121,
|
|
"description": "The length of the video."
|
|
}),
|
|
"steps": ("INT", {
|
|
"default": 15,
|
|
"description": "Number of steps to generate the video."
|
|
}),
|
|
"width": ("INT", {
|
|
"default": 768,
|
|
"description": "The width of the video."
|
|
}),
|
|
"height": ("INT", {
|
|
"default": 768,
|
|
"description": "The height of the video."
|
|
}),
|
|
"crf": ("INT", {
|
|
"default": 28,
|
|
"min": 0,
|
|
"max": 51,
|
|
"step": 1
|
|
}),
|
|
"terminal": ("FLOAT", {
|
|
"default": 0.1,
|
|
"step": 0.01,
|
|
"description": "The terminal values of the sigmas after stretching."
|
|
}),
|
|
},
|
|
"optional": {
|
|
"seed": (
|
|
"INT",
|
|
),
|
|
"user_prompt": (
|
|
"STRING",
|
|
),
|
|
"pre_prompt": (
|
|
"STRING",
|
|
),
|
|
"post_prompt": (
|
|
"STRING",
|
|
),
|
|
"negative_prompt": (
|
|
"STRING",
|
|
),
|
|
}
|
|
}
|
|
|
|
RETURN_TYPES = ("STRING", "INT", "INT", "INT", "INT", "INT", "FLOAT", "STRING", "STRING", "STRING", "STRING", "NOISE", "STRING")
|
|
RETURN_NAMES = ("image_url", "length", "steps", "width", "height", "crf", "terminal", "user_prompt", "pre_prompt", "post_prompt", "negative_prompt", "seed", "input_metadata")
|
|
FUNCTION = "load_inputs"
|
|
CATEGORY = "MemeDeck"
|
|
|
|
def load_inputs(self, image_url, length=121, steps=15, width=768, height=768, crf=28, terminal=0.1, user_prompt="", pre_prompt="", post_prompt="", negative_prompt="", seed=None):
|
|
input_metadata = json.dumps({
|
|
"length": length,
|
|
"steps": steps,
|
|
"width": width,
|
|
"height": height,
|
|
"crf": crf,
|
|
"terminal": terminal,
|
|
"user_prompt": user_prompt,
|
|
"pre_prompt": pre_prompt,
|
|
"post_prompt": post_prompt,
|
|
})
|
|
return (image_url, length, steps, width, height, crf, terminal, user_prompt, pre_prompt, post_prompt, negative_prompt, Noise_RandomNoise(seed), input_metadata)
|