diff --git a/__init__.py b/__init__.py index a764f19..dedda51 100644 --- a/__init__.py +++ b/__init__.py @@ -24,8 +24,7 @@ from .src.comfyui_ino_nodes.utils.extra_nodes import InoBoolToSwitch, InoIntEqua from .src.comfyui_ino_nodes.utils.file_helper import Zip, Unzip, RemoveFile, RemoveFolder, IncrementBatchName -from .src.comfyui_ino_nodes.utils.sampler_helper import InoGetModelConfig, InoShowModelConfig, InoUpdateModelConfig, InoGetLoraConfig, InoShowLoraConfig -from .src.comfyui_ino_nodes.utils.sampler_helper import InoLoadSamplerModels, InoGetConditioning, InoGetSamplerConfig +from .src.comfyui_ino_nodes.utils.sampler_helper import InoRandomNoise, InoGetModelConfig, InoShowModelConfig, InoUpdateModelConfig, InoGetLoraConfig, InoShowLoraConfig, InoLoadSamplerModels, InoGetConditioning, InoGetSamplerConfig from .src.comfyui_ino_nodes.node_cloudreve import CloudreveInit, CloudreveSignin, CloudreveUploadFile @@ -58,6 +57,7 @@ NODE_CLASS_MAPPINGS = { "RemoveFolder": RemoveFolder, "IncrementBatchName": IncrementBatchName, + "InoRandomNoise": InoRandomNoise, "InoGetModelConfig": InoGetModelConfig, "InoShowModelConfig": InoShowModelConfig, "InoUpdateModelConfig": InoUpdateModelConfig, @@ -102,6 +102,7 @@ NODE_DISPLAY_NAME_MAPPINGS = { "RemoveFolder": "Remove Folder", "IncrementBatchName": "Increment Batch Name", + "InoRandomNoise": "Ino Random Noise", "InoGetModelConfig": "Ino Get Model Config", "InoShowModelConfig": "Ino Show Model Config", "InoUpdateModelConfig": "Ino Update Model Config", diff --git a/src/comfyui_ino_nodes/utils/sampler_helper.py b/src/comfyui_ino_nodes/utils/sampler_helper.py index 2c779c1..5e3c547 100644 --- a/src/comfyui_ino_nodes/utils/sampler_helper.py +++ b/src/comfyui_ino_nodes/utils/sampler_helper.py @@ -119,6 +119,30 @@ def load_lora(config_str, model, clip): else: return False, model, clip, "" +from comfy_extras.nodes_custom_sampler import Noise_RandomNoise + +class InoRandomNoise: + @classmethod + def INPUT_TYPES(s): + return { + "required": { + "noise_seed": ("INT", { + "default": 0, + "min": 0, + "max": 0xffffffffffffffff, + "control_after_generate": True, + }), + } + } + + RETURN_TYPES = ("NOISE", "INT", ) + FUNCTION = "function" + CATEGORY = "InoSamplerHelper" + + def function(self, noise_seed): + random_seed = Noise_RandomNoise(noise_seed) + return (random_seed, noise_seed, ) + class InoGetModelConfig: """