Add image.py objects to available mappings

This commit is contained in:
Thomas Ward
2024-05-24 03:00:39 -04:00
parent 2358555822
commit 07701db0bb
2 changed files with 8 additions and 0 deletions
+8
View File
@@ -1,6 +1,8 @@
from .generation import TWCUI_Util_GenerationParameters, TWCUI_Util_CommonSDXLResolutions
from .generation import TWCUI_Util_GenerationPrompts
from .image import TWCUI_Util_SaveImage
from .util import TWCUI_Util_StringLiteral, TWCUI_Util_MultilineStringLiteral
from .util import TWCUI_Util_IntLiteral, TWCUI_Util_FloatLiteral
@@ -10,6 +12,9 @@ NODE_CLASS_MAPPINGS = {
"TWCUI_Util_GenerationParameters": TWCUI_Util_GenerationParameters,
"TWCUI_Util_GenerationPrompts": TWCUI_Util_GenerationPrompts,
# Image category
"TWCUI_Util_SaveImage": TWCUI_Util_SaveImage,
# Util category
"TWCUI_Util_FloatLiteral": TWCUI_Util_FloatLiteral,
"TWCUI_Util_IntLiteral": TWCUI_Util_IntLiteral,
@@ -23,6 +28,9 @@ NODE_DISPLAY_NAME_MAPPINGS = {
"TWCUI_Util_GenerationParameters": "Generation Parameters",
"TWCUI_Util_GenerationPrompts": "Prompts",
# Image category
"TWCUI_Util_SaveImage": "Save Image",
# Util category
"TWCUI_Util_FloatLiteral": "FLOAT Literal",
"TWCUI_Util_IntLiteral": "INTEGER Literal",
View File