From 417e75848a0fc4e38d9d752a7a8dd4ce3be67dc7 Mon Sep 17 00:00:00 2001 From: bedovyy Date: Sat, 1 Mar 2025 18:59:05 +0900 Subject: [PATCH] add nai-diffusion-4-full model --- nodes.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/nodes.py b/nodes.py index 8b437ca..82fa62c 100644 --- a/nodes.py +++ b/nodes.py @@ -40,7 +40,7 @@ class ModelOption: def INPUT_TYPES(s): return { "required": { - "model": (["safe-diffusion", "nai-diffusion", "nai-diffusion-furry", "nai-diffusion-2", "nai-diffusion-furry-3", "nai-diffusion-3", "nai-diffusion-4-curated-preview"], { "default": "nai-diffusion-4-curated-preview" }), + "model": (["safe-diffusion", "nai-diffusion", "nai-diffusion-furry", "nai-diffusion-2", "nai-diffusion-furry-3", "nai-diffusion-3", "nai-diffusion-4-curated-preview", "nai-diffusion-4-full"], { "default": "nai-diffusion-4-full" }), }, "optional": { "option": ("NAID_OPTION",) }, } @@ -217,7 +217,7 @@ class GenerateNAID: } } } - model = "nai-diffusion-4-curated-preview" + model = "nai-diffusion-4-full" action = "generate" if option: @@ -252,7 +252,7 @@ class GenerateNAID: retry = option["retry"] if option and "retry" in option else None if limit_opus_free: - pixel_limit = 1024*1024 if model in ("nai-diffusion-2", "nai-diffusion-furry-3", "nai-diffusion-3", "nai-diffusion-4", "nai-diffusion-4-curated-preview") else 640*640 + pixel_limit = 1024*1024 if model in ("nai-diffusion-2", "nai-diffusion-furry-3", "nai-diffusion-3", "nai-diffusion-4", "nai-diffusion-4-curated-preview", "nai-diffusion-4-full") else 640*640 if width * height > pixel_limit: max_width, max_height = calculate_resolution(pixel_limit, (width, height)) params["width"] = max_width @@ -263,7 +263,7 @@ class GenerateNAID: if variety: params["skip_cfg_above_sigma"] = calculate_skip_cfg_above_sigma(params["width"], params["height"]) - if sampler == "ddim" and model in ("nai-diffusion-3", "nai-diffusion-4", "nai-diffusion-4-curated-preview"): + if sampler == "ddim" and model in ("nai-diffusion-furry-3", "nai-diffusion-3", "nai-diffusion-4-curated-preview", "nai-diffusion-4-full"): params["sampler"] = "ddim_v3" if action == "infill" and model != "nai-diffusion-2": @@ -587,4 +587,4 @@ NODE_DISPLAY_NAME_MAPPINGS = { "DeclutterNAID": "Declutter ✒️🅝🅐🅘", "V4BasePrompt": "V4 Base Prompt ✒️🅝🅐🅘", "V4NegativePrompt": "V4 Negative Prompt ✒️🅝🅐🅘", -} \ No newline at end of file +}