nodes/bedovyy-ComfyUI_NAIDGenerator
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

add nai-diffusion-4-full model

Browse Source
This commit is contained in:
bedovyy
2025-03-01 18:59:05 +09:00
parent 8ecc2a5929
commit 417e75848a
1 changed files with 5 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
nodes.py
+5 -5
View File
@@ -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 ✒️🅝🅐🅘",
}
}