182 lines
11 KiB
Python
182 lines
11 KiB
Python
from .nodes import *
|
|
from .gguf_nodes import WarpedLoaderGGUF, WarpedClipLoaderGGUF, WarpedDualClipLoaderGGUF, WarpedWan22MergeLoaderGGUF
|
|
|
|
NODE_CLASS_MAPPINGS = {
|
|
"WarpedHunyuanMultiLoraMerge": WarpedHunyuanMultiLoraMerge,
|
|
"WarpedHunyuanMultiLoraAvgMerge": WarpedHunyuanMultiLoraAvgMerge,
|
|
"WarpedHunyuanLoraMerge": WarpedHunyuanLoraMerge,
|
|
"WarpedHunyuanMultiLoraMixer": WarpedHunyuanMultiLoraMixer,
|
|
# "WarpedHunyuanMultiLoraMixerExt": WarpedHunyuanMultiLoraMixerExt,
|
|
# "WarpedHunyuanMultiLoraMixerUlt": WarpedHunyuanMultiLoraMixerUlt,
|
|
# "WarpedHunyuanMultiLoraMixerPlatinum": WarpedHunyuanMultiLoraMixerPlatinum,
|
|
"WarpedHunyuanLoraAvgMerge": WarpedHunyuanLoraAvgMerge,
|
|
"WarpedHunyuanLoraBatchMerge": WarpedHunyuanLoraBatchMerge,
|
|
"WarpedWanLoraMerge": WarpedWanLoraMerge,
|
|
"WarpedSamplerCustomAdv": WarpedSamplerCustomAdv,
|
|
"WarpedSamplerCustomAdvLatent": WarpedSamplerCustomAdvLatent,
|
|
"WarpedSamplerCustomScripted": WarpedSamplerCustomScripted,
|
|
"WarpedFramepackSampler": WarpedFramepackSampler,
|
|
# "WarpedFramepackSamplerScripted": WarpedFramepackSamplerScripted,
|
|
"WarpedCreateSpecialImageBatch": WarpedCreateSpecialImageBatch,
|
|
"WarpedCreateEmptyImageBatch": WarpedCreateEmptyImageBatch,
|
|
"WarpedCreateEmptyLatentBatch": WarpedCreateEmptyLatentBatch,
|
|
"WarpedBundleVideoImages": WarpedBundleVideoImages,
|
|
"WarpedBundleAllVideoImages": WarpedBundleAllVideoImages,
|
|
"WarpedGetImageFromVideo": WarpedGetImageFromVideo,
|
|
"WarpedGetTwoImagesFromVideo": WarpedGetTwoImagesFromVideo,
|
|
"WarpedImageNoiseAugmentation": WarpedImageNoiseAugmentation,
|
|
"WarpedLeapfusionHunyuanI2V": WarpedLeapfusionHunyuanI2V,
|
|
"WarpedSaveAnimatedPng": WarpedSaveAnimatedPng,
|
|
"WarpedUpscaleWithModel": WarpedUpscaleWithModel,
|
|
"WarpedLoadVideosBatch": WarpedLoadVideosBatch,
|
|
"WarpedWanImageToVideo": WarpedWanImageToVideo,
|
|
"WarpedWanLoadAndEditLoraBlocks": WarpedWanLoadAndEditLoraBlocks,
|
|
"WarpedImageScaleToSide": WarpedImageScaleToSide,
|
|
"WarpedLoadLorasBatchByPrefix": WarpedLoadLorasBatchByPrefix,
|
|
"WarpedHunyuanVideoLoraLoader": WarpedHunyuanVideoLoraLoader,
|
|
"WarpedHunyuanMultiLoraLoader": WarpedHunyuanMultiLoraLoader,
|
|
"WarpedHunyuanLoraConvertKeys": WarpedHunyuanLoraConvertKeys,
|
|
# "WarpedHunyuanLoraConvertKeys2": WarpedHunyuanLoraConvertKeys2,
|
|
"WarpedLoraKeysAndMetadataReader": WarpedLoraKeysAndMetadataReader,
|
|
"WarpedHunyuanLoraConvert": WarpedHunyuanLoraConvert,
|
|
"WarpedHunyuanLoraConvertToMusubi": WarpedHunyuanLoraConvertToMusubi,
|
|
"WarpedFramepackLoraSelectBatch": WarpedFramepackLoraSelectBatch,
|
|
"WarpedFramepackMultiLoraSelect": WarpedFramepackMultiLoraSelect,
|
|
"WarpedFramepackMultiLoraSelectExt": WarpedFramepackMultiLoraSelectExt,
|
|
"WarpedCLIPLoader": WarpedCLIPLoader,
|
|
"WarpedDualCLIPLoader": WarpedDualCLIPLoader,
|
|
"WarpedDualEncoder": WarpedDualEncoder,
|
|
"WarpedSingleEncoder": WarpedSingleEncoder,
|
|
"WarpedCLIPVisionLoader": WarpedCLIPVisionLoader,
|
|
"WarpedVAELoader": WarpedVAELoader,
|
|
"WarpedLoadFramePackModel": WarpedLoadFramePackModel,
|
|
"WarpedNumericalConversion": WarpedNumericalConversion,
|
|
"WarpedLoraReSave": WarpedLoraReSave,
|
|
"WarpedCheckpointLoader": WarpedCheckpointLoader,
|
|
"WarpedReverseImageBatch": WarpedReverseImageBatch,
|
|
"WarpedMultiLoraLoader": WarpedMultiLoraLoader,
|
|
"WarpedHunyuanImageToVideo": WarpedHunyuanImageToVideo,
|
|
"WarpedBasicGuider": WarpedBasicGuider,
|
|
"WarpedDualGuider": WarpedDualGuider,
|
|
"WarpedLoaderGGUF": WarpedLoaderGGUF,
|
|
"WarpedWan22MergeLoaderGGUF": WarpedWan22MergeLoaderGGUF,
|
|
"WarpedClipLoaderGGUF": WarpedClipLoaderGGUF,
|
|
"WarpedDualClipLoaderGGUF": WarpedDualClipLoaderGGUF,
|
|
"WarpedSamplerScripts5": WarpedSamplerScripts5,
|
|
"WarpedSamplerScripts8": WarpedSamplerScripts8,
|
|
"WarpedSamplerScripts12": WarpedSamplerScripts12,
|
|
"WarpedSamplerScripts16": WarpedSamplerScripts16,
|
|
"WarpedSamplerScripts20": WarpedSamplerScripts20,
|
|
"WarpedSamplerScripts30": WarpedSamplerScripts30,
|
|
"WarpedSamplerScripts40": WarpedSamplerScripts40,
|
|
"WarpedLoadImages": WarpedLoadImages,
|
|
"WarpedSaveImageCaption": WarpedSaveImageCaption,
|
|
"WarpedModifyCaptionFile": WarpedModifyCaptionFile,
|
|
"WarpedAddToCaption": WarpedAddToCaption,
|
|
"WarpedPromptConcat": WarpedPromptConcat,
|
|
"WarpedPromptConcatExt": WarpedPromptConcatExt,
|
|
"WarpedHunyuanVideoToVideo": WarpedHunyuanVideoToVideo,
|
|
"WarpedImageLossCalc": WarpedImageLossCalc,
|
|
"WarpedUNETLoader": WarpedUNETLoader,
|
|
"WarpedUNETConverter": WarpedUNETConverter,
|
|
"WarpedHunyuanLoraDoubleBlocksSwap": WarpedHunyuanLoraDoubleBlocksSwap,
|
|
"WarpedContinueWorkflowManual": WarpedContinueWorkflowManual,
|
|
"WarpedContinueWorkflowAuto": WarpedContinueWorkflowAuto,
|
|
# "WarpedHunyuanLoraDoubleBlocksModifySegment": WarpedHunyuanLoraDoubleBlocksModifySegment,
|
|
"WarpedHunyuanLoraDoubleBlocksModifyMultipleSegments": WarpedHunyuanLoraDoubleBlocksModifyMultipleSegments,
|
|
"WarpedHunyuanLoraDoubleBlocksRemoveLinear": WarpedHunyuanLoraDoubleBlocksRemoveLinear,
|
|
"WarpedHunyuanLoraDoubleBlocksLayersBlend": WarpedHunyuanLoraDoubleBlocksLayersBlend,
|
|
"WarpedLoadHunyuanLoraWeightsByPrefix": WarpedLoadHunyuanLoraWeightsByPrefix,
|
|
}
|
|
|
|
NODE_DISPLAY_NAME_MAPPINGS = {
|
|
"WarpedHunyuanMultiLoraMerge": "Warped Hunyuan Multi Lora Merge",
|
|
"WarpedHunyuanMultiLoraAvgMerge": "Warped Hunyuan Multi Lora Avg Merge",
|
|
"WarpedHunyuanLoraMerge": "Warped Hunyuan Lora Merge",
|
|
"WarpedHunyuanMultiLoraMixer": "Warped Hunyuan Multi Lora Mixer",
|
|
# "WarpedHunyuanMultiLoraMixerExt": "Warped Hunyuan Multi Lora Mixer Ext",
|
|
# "WarpedHunyuanMultiLoraMixerUlt": "Warped Hunyuan Multi Lora Mixer Ultimate ",
|
|
# "WarpedHunyuanMultiLoraMixerPlatinum": "Warped Hunyuan Multi Lora Mixer Platinum",
|
|
"WarpedHunyuanLoraAvgMerge": "Warped Hunyuan Lora Avg Merge",
|
|
"WarpedHunyuanLoraBatchMerge": "Warped Hunyuan Lora Batch Merge",
|
|
"WarpedWanLoraMerge": "Warped Wan Lora Merge",
|
|
"WarpedSamplerCustomAdv": "Warped Sampler Custom Advanced (Image)",
|
|
"WarpedSamplerCustomAdvLatent": "Warped Sampler Custom Advanced (Latent)",
|
|
"WarpedSamplerCustomScripted": "Warped Sampler Custom Scripted",
|
|
"WarpedFramepackSampler": "Warped Framepack Sampler",
|
|
# "WarpedFramepackSamplerScripted": "Warped Framepack Sampler Scripted",
|
|
"WarpedCreateSpecialImageBatch": "Warped Create Special Image Batch",
|
|
"WarpedCreateEmptyImageBatch": "Warped Create Empty Image Batch",
|
|
"WarpedCreateEmptyLatentBatch": "Warped Create Empty Latent Batch",
|
|
"WarpedBundleVideoImages": "Warped Bundle Video Images",
|
|
"WarpedBundleAllVideoImages": "Warped Bundle All Video Images",
|
|
"WarpedGetImageFromVideo": "Warped Get Image From Video",
|
|
"WarpedGetTwoImagesFromVideo": "Warped Get Two Images From Video",
|
|
"WarpedImageNoiseAugmentation": "Warped Image Noise Augmentation",
|
|
"WarpedLeapfusionHunyuanI2V": "Warped Leapfusion Hunyuan I2V",
|
|
"WarpedSaveAnimatedPng": "Warped Save Animated Png",
|
|
"WarpedUpscaleWithModel": "Warped Upscale With Model",
|
|
"WarpedLoadVideosBatch": "Warped Load Videos Batch",
|
|
"WarpedWanImageToVideo": "Warped Wan Image To Video",
|
|
"WarpedWanLoadAndEditLoraBlocks": "Warped Wan Load And Edit Lora Blocks",
|
|
"WarpedImageScaleToSide": "Warped Image Scale To Side",
|
|
"WarpedLoadLorasBatchByPrefix": "Warped Load Loras Batch By Prefix",
|
|
"WarpedHunyuanVideoLoraLoader": "Warped Hunyuan Video Lora Loader",
|
|
"WarpedHunyuanMultiLoraLoader": "Warped Hunyuan Video Multi Lora Loader",
|
|
"WarpedHunyuanLoraConvertKeys": "Warped Hunyuan Lora Convert Keys",
|
|
# "WarpedHunyuanLoraConvertKeys2": "Warped Hunyuan Lora Convert Keys 2",
|
|
"WarpedLoraKeysAndMetadataReader": "Warped Lora Keys And Metadata Reader",
|
|
"WarpedHunyuanLoraConvert": "Warped Hunyuan Lora Convert",
|
|
"WarpedHunyuanLoraConvertToMusubi": "Warped Hunyuan Lora Convert To Musubi",
|
|
"WarpedFramepackLoraSelectBatch": "Warped Framepack Lora Select Batch",
|
|
"WarpedFramepackMultiLoraSelect": "Warped Framepack Multi Lora Select",
|
|
"WarpedFramepackMultiLoraSelectExt": "Warped Framepack Multi Lora Select Ext",
|
|
"WarpedCLIPLoader": "Warped CLIP Loader",
|
|
"WarpedDualCLIPLoader": "Warped Dual CLIP Loader",
|
|
"WarpedDualEncoder": "Warped Dual Encoder",
|
|
"WarpedSingleEncoder": "Warped Single Encoder",
|
|
"WarpedCLIPVisionLoader": "Warped Load CLIP Vision",
|
|
"WarpedVAELoader": "Warped Load VAE",
|
|
"WarpedLoadFramePackModel": "Warped Load FramePack Model",
|
|
"WarpedNumericalConversion": "Warped Numerical Conversion",
|
|
"WarpedLoraReSave": "Warped Lora Re-Save",
|
|
"WarpedCheckpointLoader": "Warped Checkpoint Loader",
|
|
"WarpedReverseImageBatch": "Warped Reverse Image Batch",
|
|
"WarpedMultiLoraLoader": "Warped Multi Lora Loader",
|
|
"WarpedHunyuanImageToVideo": "Warped Hunyuan Image To Video",
|
|
"WarpedBasicGuider": "Warped Basic Guider Batch",
|
|
"WarpedDualGuider": "Warped Dual Guider",
|
|
"WarpedLoaderGGUF": "Warped UNET Loader GGUF",
|
|
"WarpedWan22MergeLoaderGGUF": "Warped Wan22 Merge Loader GGUF",
|
|
"WarpedClipLoaderGGUF": "Warped Clip Loader GGUF",
|
|
"WarpedDualClipLoaderGGUF": "Warped Dual Clip Loader GGUF",
|
|
"WarpedSamplerScripts5": "Warped Sampler Scripts 5",
|
|
"WarpedSamplerScripts8": "Warped Sampler Scripts 8",
|
|
"WarpedSamplerScripts12": "Warped Sampler Scripts 12",
|
|
"WarpedSamplerScripts16": "Warped Sampler Scripts 16",
|
|
"WarpedSamplerScripts20": "Warped Sampler Scripts 20",
|
|
"WarpedSamplerScripts30": "Warped Sampler Scripts 30",
|
|
"WarpedSamplerScripts40": "Warped Sampler Scripts 40",
|
|
"WarpedLoadImages": "Warped Load Images",
|
|
"WarpedSaveImageCaption": "Warped Save Image Caption",
|
|
"WarpedModifyCaptionFile": "Warped Modify Caption File",
|
|
"WarpedAddToCaption": "Warped Prefix or Postfix Caption",
|
|
"WarpedPromptConcat": "Warped Prompt Concat",
|
|
"WarpedPromptConcatExt": "Warped Prompt Concat Ext",
|
|
"WarpedHunyuanVideoToVideo": "Warped Hunyuan Video To Video",
|
|
"WarpedImageLossCalc": "Warped Image Loss Calc",
|
|
"WarpedUNETLoader": "Warped Diffusion Model Loader",
|
|
"WarpedUNETConverter": "Warped Diffusion Model Converter",
|
|
"WarpedHunyuanLoraDoubleBlocksSwap": "Warped Hunyuan Lora Double Blocks Swap",
|
|
"WarpedContinueWorkflowManual": "Warped Continue Workflow Manual",
|
|
"WarpedContinueWorkflowAuto": "Warped Continue Workflow Auto",
|
|
# "WarpedHunyuanLoraDoubleBlocksModifySegment": "Warped Hunyuan Lora Double Blocks Modify Segment",
|
|
"WarpedHunyuanLoraDoubleBlocksModifyMultipleSegments": "Warped Hunyuan Lora Double Blocks Modify Multiple Segments",
|
|
"WarpedHunyuanLoraDoubleBlocksRemoveLinear": "Warped Hunyuan Lora Double Blocks Remove Linear",
|
|
"WarpedHunyuanLoraDoubleBlocksLayersBlend": "Warped Hunyuan Lora Double Blocks Layers Blend",
|
|
"WarpedLoadHunyuanLoraWeightsByPrefix": "Warped Load Hunyuan Lora Weights By Prefix",
|
|
}
|
|
|
|
WEB_DIRECTORY = "./web"
|
|
__all__ = ["NODE_CLASS_MAPPINGS", "NODE_DISPLAY_NAME_MAPPINGS", "WEB_DIRECTORY"]
|