diff --git a/__init__.py b/__init__.py index 1675983..863aecc 100644 --- a/__init__.py +++ b/__init__.py @@ -40,6 +40,7 @@ from .py.context_merge_big import RgthreeContextMergeBig from .py.image_comparer import RgthreeImageComparer from .py.power_lora_loader import RgthreePowerLoraLoader from .py.power_primitive import RgthreePowerPrimitive +from .py.image_or_latent_size import RgthreeImageOrLatentSize NODE_CLASS_MAPPINGS = { RgthreeBigContext.NAME: RgthreeBigContext, @@ -63,6 +64,7 @@ NODE_CLASS_MAPPINGS = { RgthreeImageComparer.NAME: RgthreeImageComparer, RgthreePowerLoraLoader.NAME: RgthreePowerLoraLoader, RgthreePowerPrimitive.NAME: RgthreePowerPrimitive, + RgthreeImageOrLatentSize.NAME: RgthreeImageOrLatentSize, } if get_config_value('unreleased.dynamic_context.enabled') is True: diff --git a/src_web/comfyui/constants.ts b/src_web/comfyui/constants.ts index 0b68ab4..e75e200 100644 --- a/src_web/comfyui/constants.ts +++ b/src_web/comfyui/constants.ts @@ -19,6 +19,8 @@ export const NodeTypesString = { DYNAMIC_CONTEXT: addRgthree("Dynamic Context"), DYNAMIC_CONTEXT_SWITCH: addRgthree("Dynamic Context Switch"), DISPLAY_ANY: addRgthree("Display Any"), + IMAGE_OR_LATENT_SIZE: addRgthree("Image or Latent Size"), + NODE_MODE_RELAY: addRgthree("Mute / Bypass Relay"), NODE_MODE_REPEATER: addRgthree("Mute / Bypass Repeater"), FAST_MUTER: addRgthree("Fast Muter"), diff --git a/web/comfyui/constants.js b/web/comfyui/constants.js index 666f5d4..6ef9e01 100644 --- a/web/comfyui/constants.js +++ b/web/comfyui/constants.js @@ -16,6 +16,7 @@ export const NodeTypesString = { DYNAMIC_CONTEXT: addRgthree("Dynamic Context"), DYNAMIC_CONTEXT_SWITCH: addRgthree("Dynamic Context Switch"), DISPLAY_ANY: addRgthree("Display Any"), + IMAGE_OR_LATENT_SIZE: addRgthree("Image or Latent Size"), NODE_MODE_RELAY: addRgthree("Mute / Bypass Relay"), NODE_MODE_REPEATER: addRgthree("Mute / Bypass Repeater"), FAST_MUTER: addRgthree("Fast Muter"),