150 lines
8.7 KiB
Python
150 lines
8.7 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,
|
|
"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,
|
|
}
|
|
|
|
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 ",
|
|
"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",
|
|
}
|
|
|
|
WEB_DIRECTORY = "./web"
|
|
__all__ = ["NODE_CLASS_MAPPINGS", "NODE_DISPLAY_NAME_MAPPINGS", "WEB_DIRECTORY"]
|