diff --git a/nodes.py b/nodes.py index 27ff18a..6bbf36d 100644 --- a/nodes.py +++ b/nodes.py @@ -38,7 +38,7 @@ class ModelOption: def INPUT_TYPES(s): return { "required": { - "model": (["safe-diffusion", "nai-diffusion", "nai-diffusion-furry", "nai-diffusion-2", "nai-diffusion-3"], { "default": "nai-diffusion-3" }), + "model": (["safe-diffusion", "nai-diffusion", "nai-diffusion-furry", "nai-diffusion-2", "nai-diffusion-furry-3", "nai-diffusion-3"], { "default": "nai-diffusion-3" }), }, "optional": { "option": ("NAID_OPTION",) }, } @@ -226,7 +226,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-3",) else 640*640 + pixel_limit = 1024*1024 if model in ("nai-diffusion-2", "nai-diffusion-furry-3", "nai-diffusion-3",) else 640*640 if width * height > pixel_limit: max_width, max_height = calculate_resolution(pixel_limit, (width, height)) params["width"] = max_width diff --git a/pyproject.toml b/pyproject.toml index c52d807..042968a 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -1,7 +1,7 @@ [project] name = "comfyui_naidgenerator" description = "This extension helps generate images through NAI." -version = "1.0.3" +version = "1.0.4" license = { file = "LICENSE" } dependencies = ["python-dotenv", "argon2-cffi"]