From 3646b7c315b871679ed3da023b07f6e649ea0085 Mon Sep 17 00:00:00 2001 From: DreamFly Date: Mon, 2 Jun 2025 23:21:27 +0800 Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E6=94=B9=E5=A4=9A=E9=9A=8F=E6=9C=BA?= =?UTF-8?q?=E7=94=9F=E6=88=90=E6=8F=90=E7=A4=BA=E8=AF=8D?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- nodes/prompt_randoms.py | 36 +++++++++++++++++++++++++++++++++--- 1 file changed, 33 insertions(+), 3 deletions(-) diff --git a/nodes/prompt_randoms.py b/nodes/prompt_randoms.py index d6b0c03..3e1e557 100644 --- a/nodes/prompt_randoms.py +++ b/nodes/prompt_randoms.py @@ -35,6 +35,12 @@ class RandomsPrompt: "classify6": (buildOptionList(config.classifies, True, False), { "default": DEFAULT_OPTION }), + "classify7": (buildOptionList(config.classifies, True, False), { + "default": DEFAULT_OPTION + }), + "classify8": (buildOptionList(config.classifies, True, False), { + "default": DEFAULT_OPTION + }), "seed": ("INT", { "default": 0, "min": 0, "max": 0xFFFFFFFFFFFFFFFF }), # 启用 "enable": ("BOOLEAN", {"default": True}) @@ -49,7 +55,7 @@ class RandomsPrompt: FUNCTION = "randoms_prompt" CATEGORY = "PromptWrapper" - def randoms_prompt(self, language, classify1, classify2, classify3, classify4, classify5, classify6, + def randoms_prompt(self, language, classify1, classify2, classify3, classify4, classify5, classify6, classify7, classify8, seed=0, enable=True, input_prompt=""): prompt_words = [] @@ -61,7 +67,7 @@ class RandomsPrompt: # 获取Script的目录路径 script_dir = os.path.dirname(os.path.abspath(__file__)) # Script directory language_dir = "zh" if language == "Chinese" else "en" - classifys = [classify1, classify2, classify3, classify4, classify5, classify6] + classifys = [classify1, classify2, classify3, classify4, classify5, classify6, classify7, classify8] for classify in classifys: if classify != EMPTY_OPTION: @@ -164,6 +170,28 @@ class RandomsWeightPrompt: "step": 0.05, "display": "slider", }), + "classify7": (buildOptionList(config.classifies, True, False), { + "default": DEFAULT_OPTION + }), + "classify7_weight": ("FLOAT", { + "default": 1, + "min": 0, + "max": max_float_value, + "round": 0.01, + "step": 0.05, + "display": "slider", + }), + "classify8": (buildOptionList(config.classifies, True, False), { + "default": DEFAULT_OPTION + }), + "classify8_weight": ("FLOAT", { + "default": 1, + "min": 0, + "max": max_float_value, + "round": 0.01, + "step": 0.05, + "display": "slider", + }), "seed": ("INT", { "default": 0, "min": 0, "max": 0xFFFFFFFFFFFFFFFF }), # 启用 "enable": ("BOOLEAN", {"default": True}) @@ -180,6 +208,7 @@ class RandomsWeightPrompt: def randoms_weight_prompt(self, language, classify1, classify1_weight, classify2, classify2_weight, classify3, classify3_weight, classify4, classify4_weight, classify5, classify5_weight, classify6, classify6_weight, + classify7, classify7_weight, classify8, classify8_weight, seed=0, enable=True, input_prompt=""): prompt_words = [] @@ -192,7 +221,8 @@ class RandomsWeightPrompt: script_dir = os.path.dirname(os.path.abspath(__file__)) # Script directory language_dir = "zh" if language == "Chinese" else "en" classify_weights = [(classify1, classify1_weight), (classify2, classify2_weight), (classify3, classify3_weight), - (classify4, classify4_weight), (classify5, classify5_weight), (classify6, classify6_weight)] + (classify4, classify4_weight), (classify5, classify5_weight), (classify6, classify6_weight), + (classify7, classify7_weight), (classify8, classify8_weight)] for classify_weight in classify_weights: classify = classify_weight[0]