110 lines
4.2 KiB
Python
110 lines
4.2 KiB
Python
from .utils import (
|
|
CastAnyToIntStringFloat,
|
|
ConditionalPassThrough,
|
|
StringListIndex,
|
|
StringListStepper,
|
|
StringListRandom,
|
|
StringJoiner,
|
|
RegexFilter,
|
|
DelimiterSplit,
|
|
IndexWrap,
|
|
AdvancedCalculator,
|
|
SwitchAny,
|
|
SwitchAnyByIndex,
|
|
)
|
|
from .images import (
|
|
JoinImageBatch,
|
|
ValidateImageShape,
|
|
AppendImageBatch,
|
|
SwitchImageByIndex,
|
|
VantageMultiImageLoader,
|
|
VantageUnbatchImages,
|
|
VantageImagesOrNone,
|
|
)
|
|
from .latents import (
|
|
JoinLatentBatch,
|
|
AppendLatentBatch,
|
|
SwitchLatentByIndex,
|
|
)
|
|
from .block_swap import VantageWanBlockSwap
|
|
from .gguf import VantageGGUFLoader
|
|
from .prompt_scene_router import PromptSceneRouter
|
|
from .qwen_tts_downloader import QwenTTSModelDownloader
|
|
from .qwen_tts_voice_design import QwenTTSVoiceDesignNode
|
|
from .qwen_tts_voice_clone_prompt import QwenTTSVoiceClonePromptNode
|
|
from .qwen_tts_speaker_bank import QwenTTSSpeakerBankNode
|
|
from .qwen_tts_multi_speaker import QwenTTSMultiSpeakerNode
|
|
from .krea2 import VantageTextEncodeKrea2
|
|
|
|
NODE_CLASS_MAPPINGS = {
|
|
"CastAnyToIntStringFloat": CastAnyToIntStringFloat,
|
|
"ConditionalPassThrough": ConditionalPassThrough,
|
|
"StringListIndex": StringListIndex,
|
|
"StringListStepper": StringListStepper,
|
|
"StringListRandom": StringListRandom,
|
|
"StringJoiner": StringJoiner,
|
|
"RegexFilter": RegexFilter,
|
|
"DelimiterSplit": DelimiterSplit,
|
|
"IndexWrap": IndexWrap,
|
|
"AdvancedCalculator": AdvancedCalculator,
|
|
"SwitchAny": SwitchAny,
|
|
"SwitchAnyByIndex": SwitchAnyByIndex,
|
|
"JoinImageBatch": JoinImageBatch,
|
|
"ValidateImageShape": ValidateImageShape,
|
|
"AppendImageBatch": AppendImageBatch,
|
|
"SwitchImageByIndex": SwitchImageByIndex,
|
|
"JoinLatentBatch": JoinLatentBatch,
|
|
"AppendLatentBatch": AppendLatentBatch,
|
|
"SwitchLatentByIndex": SwitchLatentByIndex,
|
|
"VantageWanBlockSwap": VantageWanBlockSwap,
|
|
"VantageGGUFLoader": VantageGGUFLoader,
|
|
"PromptSceneRouter": PromptSceneRouter,
|
|
"QwenTTSModelDownloader": QwenTTSModelDownloader,
|
|
"QwenTTSVoiceDesignNode": QwenTTSVoiceDesignNode,
|
|
"QwenTTSVoiceClonePromptNode": QwenTTSVoiceClonePromptNode,
|
|
"QwenTTSSpeakerBankNode": QwenTTSSpeakerBankNode,
|
|
"QwenTTSMultiSpeakerNode": QwenTTSMultiSpeakerNode,
|
|
"VantageTextEncodeKrea2": VantageTextEncodeKrea2,
|
|
"VantageMultiImageLoader": VantageMultiImageLoader,
|
|
"VantageUnbatchImages": VantageUnbatchImages,
|
|
"VantageImagesOrNone": VantageImagesOrNone,
|
|
}
|
|
|
|
NODE_DISPLAY_NAME_MAPPINGS = {
|
|
"CastAnyToIntStringFloat": "Cast Any → INT / STRING / FLOAT",
|
|
"ConditionalPassThrough": "Conditional Pass Through",
|
|
"StringListIndex": "Multiline String → Item & Count",
|
|
"StringListStepper": "String List Stepper",
|
|
"StringListRandom": "Random String From List",
|
|
"StringJoiner": "Join Multiline String",
|
|
"RegexFilter": "Regex Filter (Multiline)",
|
|
"DelimiterSplit": "Delimiter → Multiline Split",
|
|
"IndexWrap": "Index Wrap / Clamp",
|
|
"AdvancedCalculator": "Advanced Calculator (Expression)",
|
|
"SwitchAny": "Switch Any",
|
|
"SwitchAnyByIndex": "Switch Any (By Index)",
|
|
"JoinImageBatch": "Join Image Batch",
|
|
"ValidateImageShape": "Validate Image Shape",
|
|
"AppendImageBatch": "Append Image Batch",
|
|
"SwitchImageByIndex": "Switch Image (By Index)",
|
|
"JoinLatentBatch": "Join Latent Batch",
|
|
"AppendLatentBatch": "Append Latent Batch",
|
|
"SwitchLatentByIndex": "Switch Latent (By Index)",
|
|
"VantageWanBlockSwap": "Vantage Wan Video Block Swap",
|
|
"VantageGGUFLoader": "Vantage GGUF UNET Loader",
|
|
"PromptSceneRouter": "Vantage Prompt Scene Router",
|
|
"QwenTTSModelDownloader": "(Down)load Qwen TTS Models",
|
|
"QwenTTSVoiceDesignNode": "Qwen TTS Voice Design Node",
|
|
"QwenTTSVoiceClonePromptNode": "Qwen TTS Voice Clone Prompt",
|
|
"QwenTTSSpeakerBankNode": "Qwen TTS Speaker Bank",
|
|
"QwenTTSMultiSpeakerNode": "Qwen TTS Multi Speaker Node",
|
|
"VantageTextEncodeKrea2": "Vantage Text Encode (Krea2)",
|
|
"VantageMultiImageLoader": "Vantage Multiple Image Loader",
|
|
"VantageUnbatchImages": "Vantage Unbatch Images",
|
|
"VantageImagesOrNone": "Vantage Images or None",
|
|
}
|
|
|
|
WEB_DIRECTORY = "./web"
|
|
|
|
__all__ = ["NODE_CLASS_MAPPINGS", "NODE_DISPLAY_NAME_MAPPINGS", "WEB_DIRECTORY"]
|