|
|
@ -1018,6 +1018,54 @@ NODE_CLASS_MAPPINGS = { |
|
|
|
"VAEEncodeTiled": VAEEncodeTiled, |
|
|
|
"VAEEncodeTiled": VAEEncodeTiled, |
|
|
|
} |
|
|
|
} |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
NODE_DISPLAY_NAME_MAPPINGS = { |
|
|
|
|
|
|
|
# Sampling |
|
|
|
|
|
|
|
"KSampler": "KSampler", |
|
|
|
|
|
|
|
"KSamplerAdvanced": "KSampler (Advanced)", |
|
|
|
|
|
|
|
# Loaders |
|
|
|
|
|
|
|
"CheckpointLoader": "Load Checkpoint", |
|
|
|
|
|
|
|
"CheckpointLoaderSimple": "Load Checkpoint (Simple)", |
|
|
|
|
|
|
|
"VAELoader": "Load VAE", |
|
|
|
|
|
|
|
"LoraLoader": "Load LoRA", |
|
|
|
|
|
|
|
"CLIPLoader": "Load CLIP", |
|
|
|
|
|
|
|
"ControlNetLoader": "Load ControlNet Model", |
|
|
|
|
|
|
|
"DiffControlNetLoader": "Load ControlNet Model (diff)", |
|
|
|
|
|
|
|
"StyleModelLoader": "Load Style Model", |
|
|
|
|
|
|
|
"CLIPVisionLoader": "Load CLIP Vision", |
|
|
|
|
|
|
|
"UpscaleModelLoader": "Load Upscale Model", |
|
|
|
|
|
|
|
# Conditioning |
|
|
|
|
|
|
|
"CLIPVisionEncode": "CLIP Vision Encode", |
|
|
|
|
|
|
|
"StyleModelApply": "Apply Style Model", |
|
|
|
|
|
|
|
"CLIPTextEncode": "CLIP Text Encode (Prompt)", |
|
|
|
|
|
|
|
"CLIPSetLastLayer": "CLIP Set Last Layer", |
|
|
|
|
|
|
|
"ConditioningCombine": "Conditioning (Combine)", |
|
|
|
|
|
|
|
"ConditioningSetArea": "Conditioning (Set Area)", |
|
|
|
|
|
|
|
"ControlNetApply": "Apply ControlNet", |
|
|
|
|
|
|
|
# Latent |
|
|
|
|
|
|
|
"VAEEncodeForInpaint": "VAE Encode (for Inpainting)", |
|
|
|
|
|
|
|
"SetLatentNoiseMask": "Set Latent Noise Mask", |
|
|
|
|
|
|
|
"VAEDecode": "VAE Decode", |
|
|
|
|
|
|
|
"VAEEncode": "VAE Encode", |
|
|
|
|
|
|
|
"LatentRotate": "Rotate Latent", |
|
|
|
|
|
|
|
"LatentFlip": "Flip Latent", |
|
|
|
|
|
|
|
"LatentCrop": "Crop Latent", |
|
|
|
|
|
|
|
"EmptyLatentImage": "Empty Latent Image", |
|
|
|
|
|
|
|
"LatentUpscale": "Upscale Latent", |
|
|
|
|
|
|
|
"LatentComposite": "Latent Composite", |
|
|
|
|
|
|
|
# Image |
|
|
|
|
|
|
|
"SaveImage": "Save Image", |
|
|
|
|
|
|
|
"PreviewImage": "Preview Image", |
|
|
|
|
|
|
|
"LoadImage": "Load Image", |
|
|
|
|
|
|
|
"LoadImageMask": "Load Image (as Mask)", |
|
|
|
|
|
|
|
"ImageScale": "Upscale Image", |
|
|
|
|
|
|
|
"ImageUpscaleWithModel": "Upscale Image (using Model)", |
|
|
|
|
|
|
|
"ImageInvert": "Invert Image", |
|
|
|
|
|
|
|
"ImagePadForOutpaint": "Pad Image for Outpainting", |
|
|
|
|
|
|
|
# _for_testing |
|
|
|
|
|
|
|
"VAEDecodeTiled": "VAE Decode (Tiled)", |
|
|
|
|
|
|
|
"VAEEncodeTiled": "VAE Encode (Tiled)", |
|
|
|
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
|
|
def load_custom_node(module_path): |
|
|
|
def load_custom_node(module_path): |
|
|
|
module_name = os.path.basename(module_path) |
|
|
|
module_name = os.path.basename(module_path) |
|
|
|
if os.path.isfile(module_path): |
|
|
|
if os.path.isfile(module_path): |
|
|
|