Add ModelVAELORALoader to mappings

This commit is contained in:
Thomas Ward
2024-05-28 14:19:03 -04:00
parent d90740a33e
commit 4f2530b50e
+3
View File
@@ -1,5 +1,6 @@
from .generation import TWCUI_Util_GenerationParameters, TWCUI_Util_CommonSDXLResolutions
from .generation import TWCUI_Util_GenerationPrompts, TWCUI_Util_ModelVAELoader
from .generation import TWCUI_Util_ModelVAELORALoader
from .image import TWCUI_Util_SaveImage, TWCUI_Util_SaveImageAdvanced
@@ -9,6 +10,7 @@ from .util import TWCUI_Util_IntLiteral, TWCUI_Util_FloatLiteral
NODE_CLASS_MAPPINGS = {
# Generation category
"TWCUI_Util_ModelVAELoader": TWCUI_Util_ModelVAELoader,
"TWCUI_Util_ModelVAELORALoader": TWCUI_Util_ModelVAELORALoader,
"TWCUI_Util_CommonSDXLResolutions": TWCUI_Util_CommonSDXLResolutions,
"TWCUI_Util_GenerationParameters": TWCUI_Util_GenerationParameters,
"TWCUI_Util_GenerationPrompts": TWCUI_Util_GenerationPrompts,
@@ -27,6 +29,7 @@ NODE_CLASS_MAPPINGS = {
NODE_DISPLAY_NAME_MAPPINGS = {
# Generation category
"TWCUI_Util_ModelVAELoader": "Checkpoint / VAE Loader",
"TWCUI_Util_ModelVAELORALoader": "Checkpoint / VAE / LoRA Loader",
"TWCUI_Util_CommonSDXLResolutions": "Common SDXL Resolutions",
"TWCUI_Util_GenerationParameters": "Generation Parameters",
"TWCUI_Util_GenerationPrompts": "Prompts",