Fix missing item in returns, update names.

This commit is contained in:
Thomas Ward
2024-05-24 01:26:29 -04:00
parent 1b81f40548
commit afb10a2b1c
+3 -2
View File
@@ -119,9 +119,10 @@ class TWCUI_Util_GenerationParameters(BaseNode):
}
}
RETURN_TYPES = ("MODEL", "CLIP", "VAE", "INT", "INT", "FLOAT",
RETURN_TYPES = ("MODEL", "CLIP", "VAE", "INT", "INT", "INT", "FLOAT",
comfy.samplers.KSampler.SAMPLERS, comfy.samplers.KSampler.SCHEDULERS, "INT")
RETURN_NAMES = ("MODEL", "CLIP", "VAE", "width", "height", "cfg_scale", "sampler_name", "scheduler_name", "seed")
RETURN_NAMES = ("MODEL", "CLIP", "VAE", "width", "height", "steps", "cfg_scale", "sampler_name", "scheduler",
"seed")
def process(self, ckpt_name: str, vae_name: str, image_width: int, image_height: int, sampling_steps: int,
cfg_scale: float, sampler_name: str, scheduler_name: str, seed: int) -> tuple: