149 lines
12 KiB
Python
149 lines
12 KiB
Python
from .nodes.load_image_with_subfolders import NODE_CLASS_MAPPINGS as LOAD_IMAGE_CLASS_MAPPINGS
|
|
from .nodes.load_image_with_subfolders import NODE_DISPLAY_NAME_MAPPINGS as LOAD_IMAGE_DISPLAY_MAPPINGS
|
|
from .nodes.checkpoint_cycler_loader import NODE_CLASS_MAPPINGS as CHECKPOINT_CLASS_MAPPINGS
|
|
from .nodes.checkpoint_cycler_loader import NODE_DISPLAY_NAME_MAPPINGS as CHECKPOINT_DISPLAY_MAPPINGS
|
|
from .nodes.diffusion_clip_vae_cycler_loader import NODE_CLASS_MAPPINGS as DIFFUSION_CLIP_VAE_CLASS_MAPPINGS
|
|
from .nodes.diffusion_clip_vae_cycler_loader import NODE_DISPLAY_NAME_MAPPINGS as DIFFUSION_CLIP_VAE_DISPLAY_MAPPINGS
|
|
from .nodes.four_prompt_model_cycler_loader import NODE_CLASS_MAPPINGS as FOUR_PROMPT_MODEL_CLASS_MAPPINGS
|
|
from .nodes.four_prompt_model_cycler_loader import NODE_DISPLAY_NAME_MAPPINGS as FOUR_PROMPT_MODEL_DISPLAY_MAPPINGS
|
|
from .nodes.four_prompt_checkpoint_cycler_loader import NODE_CLASS_MAPPINGS as FOUR_PROMPT_CHECKPOINT_CLASS_MAPPINGS
|
|
from .nodes.four_prompt_checkpoint_cycler_loader import NODE_DISPLAY_NAME_MAPPINGS as FOUR_PROMPT_CHECKPOINT_DISPLAY_MAPPINGS
|
|
from .nodes.incrementing_int_string import NODE_CLASS_MAPPINGS as INT_STRING_CLASS_MAPPINGS
|
|
from .nodes.incrementing_int_string import NODE_DISPLAY_NAME_MAPPINGS as INT_STRING_DISPLAY_MAPPINGS
|
|
from .nodes.random_vocab_string_list import NODE_CLASS_MAPPINGS as RANDOM_VOCAB_STRING_LIST_CLASS_MAPPINGS
|
|
from .nodes.random_vocab_string_list import NODE_DISPLAY_NAME_MAPPINGS as RANDOM_VOCAB_STRING_LIST_DISPLAY_MAPPINGS
|
|
from .nodes.minimax_h3_prompt_builder import NODE_CLASS_MAPPINGS as MINIMAX_H3_PROMPT_BUILDER_CLASS_MAPPINGS
|
|
from .nodes.minimax_h3_prompt_builder import NODE_DISPLAY_NAME_MAPPINGS as MINIMAX_H3_PROMPT_BUILDER_DISPLAY_MAPPINGS
|
|
from .nodes.ace_song_variation_director import NODE_CLASS_MAPPINGS as ACE_SONG_VARIATION_DIRECTOR_CLASS_MAPPINGS
|
|
from .nodes.ace_song_variation_director import NODE_DISPLAY_NAME_MAPPINGS as ACE_SONG_VARIATION_DIRECTOR_DISPLAY_MAPPINGS
|
|
from .nodes.ace_song_timeline_conditioning import NODE_CLASS_MAPPINGS as ACE_SONG_TIMELINE_CLASS_MAPPINGS
|
|
from .nodes.ace_song_timeline_conditioning import NODE_DISPLAY_NAME_MAPPINGS as ACE_SONG_TIMELINE_DISPLAY_MAPPINGS
|
|
from .nodes.audio_timeline_mixer import NODE_CLASS_MAPPINGS as AUDIO_TIMELINE_MIXER_CLASS_MAPPINGS
|
|
from .nodes.audio_timeline_mixer import NODE_DISPLAY_NAME_MAPPINGS as AUDIO_TIMELINE_MIXER_DISPLAY_MAPPINGS
|
|
from .nodes.ollama_prompt_refiner import NODE_CLASS_MAPPINGS as OLLAMA_PROMPT_REFINER_CLASS_MAPPINGS
|
|
from .nodes.ollama_prompt_refiner import NODE_DISPLAY_NAME_MAPPINGS as OLLAMA_PROMPT_REFINER_DISPLAY_MAPPINGS
|
|
from .nodes.ollama_video_prompt_refiner import NODE_CLASS_MAPPINGS as OLLAMA_VIDEO_PROMPT_REFINER_CLASS_MAPPINGS
|
|
from .nodes.ollama_video_prompt_refiner import NODE_DISPLAY_NAME_MAPPINGS as OLLAMA_VIDEO_PROMPT_REFINER_DISPLAY_MAPPINGS
|
|
from .nodes.ollama_vision_captioner import NODE_CLASS_MAPPINGS as OLLAMA_VISION_CAPTIONER_CLASS_MAPPINGS
|
|
from .nodes.ollama_vision_captioner import NODE_DISPLAY_NAME_MAPPINGS as OLLAMA_VISION_CAPTIONER_DISPLAY_MAPPINGS
|
|
from .nodes.t5_token_balancer import NODE_CLASS_MAPPINGS as T5_CLASS_MAPPINGS
|
|
from .nodes.t5_token_balancer import NODE_DISPLAY_NAME_MAPPINGS as T5_DISPLAY_MAPPINGS
|
|
from .nodes.t5_equal_length_balancer import NODE_CLASS_MAPPINGS as T5_EQUAL_LENGTH_CLASS_MAPPINGS
|
|
from .nodes.t5_equal_length_balancer import NODE_DISPLAY_NAME_MAPPINGS as T5_EQUAL_LENGTH_DISPLAY_MAPPINGS
|
|
from .nodes.t5_sculpt_equal_length_balancer import NODE_CLASS_MAPPINGS as T5_SCULPT_EQUAL_LENGTH_CLASS_MAPPINGS
|
|
from .nodes.t5_sculpt_equal_length_balancer import NODE_DISPLAY_NAME_MAPPINGS as T5_SCULPT_EQUAL_LENGTH_DISPLAY_MAPPINGS
|
|
from .nodes.clip_sculpt_text_encode import NODE_CLASS_MAPPINGS as CLIP_SCULPT_TEXT_ENCODE_CLASS_MAPPINGS
|
|
from .nodes.clip_sculpt_text_encode import NODE_DISPLAY_NAME_MAPPINGS as CLIP_SCULPT_TEXT_ENCODE_DISPLAY_MAPPINGS
|
|
from .nodes.conditioning_vector_ops import NODE_CLASS_MAPPINGS as CONDITIONING_VECTOR_OPS_CLASS_MAPPINGS
|
|
from .nodes.conditioning_vector_ops import NODE_DISPLAY_NAME_MAPPINGS as CONDITIONING_VECTOR_OPS_DISPLAY_MAPPINGS
|
|
from .nodes.conditioning_tools import NODE_CLASS_MAPPINGS as CONDITIONING_TOOLS_CLASS_MAPPINGS
|
|
from .nodes.conditioning_tools import NODE_DISPLAY_NAME_MAPPINGS as CONDITIONING_TOOLS_DISPLAY_MAPPINGS
|
|
from .nodes.regional_split_regions import NODE_CLASS_MAPPINGS as REGIONAL_SPLIT_CLASS_MAPPINGS
|
|
from .nodes.regional_split_regions import NODE_DISPLAY_NAME_MAPPINGS as REGIONAL_SPLIT_DISPLAY_MAPPINGS
|
|
from .nodes.regional_prompt_encoder import NODE_CLASS_MAPPINGS as REGIONAL_PROMPT_CLASS_MAPPINGS
|
|
from .nodes.regional_prompt_encoder import NODE_DISPLAY_NAME_MAPPINGS as REGIONAL_PROMPT_DISPLAY_MAPPINGS
|
|
from .nodes.regional_sculpt_prompt_encoder import NODE_CLASS_MAPPINGS as REGIONAL_SCULPT_PROMPT_CLASS_MAPPINGS
|
|
from .nodes.regional_sculpt_prompt_encoder import NODE_DISPLAY_NAME_MAPPINGS as REGIONAL_SCULPT_PROMPT_DISPLAY_MAPPINGS
|
|
from .nodes.websocket_image_save import NODE_CLASS_MAPPINGS as WEBSOCKET_CLASS_MAPPINGS
|
|
from .nodes.websocket_image_save import NODE_DISPLAY_NAME_MAPPINGS as WEBSOCKET_DISPLAY_MAPPINGS
|
|
from .nodes.advanced_noise_sampler import NODE_CLASS_MAPPINGS as ADVANCED_NOISE_SAMPLER_CLASS_MAPPINGS
|
|
from .nodes.advanced_noise_sampler import NODE_DISPLAY_NAME_MAPPINGS as ADVANCED_NOISE_SAMPLER_DISPLAY_MAPPINGS
|
|
from .nodes.illustrious_noise_sampler import NODE_CLASS_MAPPINGS as ILLUSTRIOUS_NOISE_SAMPLER_CLASS_MAPPINGS
|
|
from .nodes.illustrious_noise_sampler import NODE_DISPLAY_NAME_MAPPINGS as ILLUSTRIOUS_NOISE_SAMPLER_DISPLAY_MAPPINGS
|
|
from .nodes.pony_v7_noise_sampler import NODE_CLASS_MAPPINGS as PONY_V7_NOISE_SAMPLER_CLASS_MAPPINGS
|
|
from .nodes.pony_v7_noise_sampler import NODE_DISPLAY_NAME_MAPPINGS as PONY_V7_NOISE_SAMPLER_DISPLAY_MAPPINGS
|
|
from .nodes.universal_noise_sampler import NODE_CLASS_MAPPINGS as UNIVERSAL_NOISE_SAMPLER_CLASS_MAPPINGS
|
|
from .nodes.universal_noise_sampler import NODE_DISPLAY_NAME_MAPPINGS as UNIVERSAL_NOISE_SAMPLER_DISPLAY_MAPPINGS
|
|
from .nodes.noise_profile_cycler import NODE_CLASS_MAPPINGS as NOISE_PROFILE_CYCLER_CLASS_MAPPINGS
|
|
from .nodes.noise_profile_cycler import NODE_DISPLAY_NAME_MAPPINGS as NOISE_PROFILE_CYCLER_DISPLAY_MAPPINGS
|
|
from .nodes.speed_sampler import NODE_CLASS_MAPPINGS as SPEED_SAMPLER_CLASS_MAPPINGS
|
|
from .nodes.speed_sampler import NODE_DISPLAY_NAME_MAPPINGS as SPEED_SAMPLER_DISPLAY_MAPPINGS
|
|
from .nodes.hiresfix_tiled import NODE_CLASS_MAPPINGS as HIRESFIX_TILED_CLASS_MAPPINGS
|
|
from .nodes.hiresfix_tiled import NODE_DISPLAY_NAME_MAPPINGS as HIRESFIX_TILED_DISPLAY_MAPPINGS
|
|
from .nodes.tiled_hiresfix_advanced import NODE_CLASS_MAPPINGS as TILED_HIRESFIX_ADVANCED_CLASS_MAPPINGS
|
|
from .nodes.tiled_hiresfix_advanced import NODE_DISPLAY_NAME_MAPPINGS as TILED_HIRESFIX_ADVANCED_DISPLAY_MAPPINGS
|
|
from .nodes.pixel_anchored_remaster import NODE_CLASS_MAPPINGS as PIXEL_ANCHORED_REMASTER_CLASS_MAPPINGS
|
|
from .nodes.pixel_anchored_remaster import NODE_DISPLAY_NAME_MAPPINGS as PIXEL_ANCHORED_REMASTER_DISPLAY_MAPPINGS
|
|
from .nodes.unet_block_noise_patch import NODE_CLASS_MAPPINGS as UNET_BLOCK_NOISE_PATCH_CLASS_MAPPINGS
|
|
from .nodes.unet_block_noise_patch import NODE_DISPLAY_NAME_MAPPINGS as UNET_BLOCK_NOISE_PATCH_DISPLAY_MAPPINGS
|
|
from .nodes.wan22_video_toolkit import NODE_CLASS_MAPPINGS as WAN22_VIDEO_CLASS_MAPPINGS
|
|
from .nodes.wan22_video_toolkit import NODE_DISPLAY_NAME_MAPPINGS as WAN22_VIDEO_DISPLAY_MAPPINGS
|
|
|
|
WEB_DIRECTORY = "./web"
|
|
|
|
NODE_CLASS_MAPPINGS = {}
|
|
NODE_CLASS_MAPPINGS.update(CHECKPOINT_CLASS_MAPPINGS)
|
|
NODE_CLASS_MAPPINGS.update(DIFFUSION_CLIP_VAE_CLASS_MAPPINGS)
|
|
NODE_CLASS_MAPPINGS.update(FOUR_PROMPT_MODEL_CLASS_MAPPINGS)
|
|
NODE_CLASS_MAPPINGS.update(FOUR_PROMPT_CHECKPOINT_CLASS_MAPPINGS)
|
|
NODE_CLASS_MAPPINGS.update(INT_STRING_CLASS_MAPPINGS)
|
|
NODE_CLASS_MAPPINGS.update(RANDOM_VOCAB_STRING_LIST_CLASS_MAPPINGS)
|
|
NODE_CLASS_MAPPINGS.update(MINIMAX_H3_PROMPT_BUILDER_CLASS_MAPPINGS)
|
|
NODE_CLASS_MAPPINGS.update(ACE_SONG_VARIATION_DIRECTOR_CLASS_MAPPINGS)
|
|
NODE_CLASS_MAPPINGS.update(ACE_SONG_TIMELINE_CLASS_MAPPINGS)
|
|
NODE_CLASS_MAPPINGS.update(AUDIO_TIMELINE_MIXER_CLASS_MAPPINGS)
|
|
NODE_CLASS_MAPPINGS.update(OLLAMA_PROMPT_REFINER_CLASS_MAPPINGS)
|
|
NODE_CLASS_MAPPINGS.update(OLLAMA_VIDEO_PROMPT_REFINER_CLASS_MAPPINGS)
|
|
NODE_CLASS_MAPPINGS.update(OLLAMA_VISION_CAPTIONER_CLASS_MAPPINGS)
|
|
NODE_CLASS_MAPPINGS.update(LOAD_IMAGE_CLASS_MAPPINGS)
|
|
NODE_CLASS_MAPPINGS.update(T5_CLASS_MAPPINGS)
|
|
NODE_CLASS_MAPPINGS.update(T5_EQUAL_LENGTH_CLASS_MAPPINGS)
|
|
NODE_CLASS_MAPPINGS.update(T5_SCULPT_EQUAL_LENGTH_CLASS_MAPPINGS)
|
|
NODE_CLASS_MAPPINGS.update(CLIP_SCULPT_TEXT_ENCODE_CLASS_MAPPINGS)
|
|
NODE_CLASS_MAPPINGS.update(CONDITIONING_VECTOR_OPS_CLASS_MAPPINGS)
|
|
NODE_CLASS_MAPPINGS.update(CONDITIONING_TOOLS_CLASS_MAPPINGS)
|
|
NODE_CLASS_MAPPINGS.update(REGIONAL_SPLIT_CLASS_MAPPINGS)
|
|
NODE_CLASS_MAPPINGS.update(REGIONAL_PROMPT_CLASS_MAPPINGS)
|
|
NODE_CLASS_MAPPINGS.update(REGIONAL_SCULPT_PROMPT_CLASS_MAPPINGS)
|
|
NODE_CLASS_MAPPINGS.update(WEBSOCKET_CLASS_MAPPINGS)
|
|
NODE_CLASS_MAPPINGS.update(ADVANCED_NOISE_SAMPLER_CLASS_MAPPINGS)
|
|
NODE_CLASS_MAPPINGS.update(ILLUSTRIOUS_NOISE_SAMPLER_CLASS_MAPPINGS)
|
|
NODE_CLASS_MAPPINGS.update(PONY_V7_NOISE_SAMPLER_CLASS_MAPPINGS)
|
|
NODE_CLASS_MAPPINGS.update(UNIVERSAL_NOISE_SAMPLER_CLASS_MAPPINGS)
|
|
NODE_CLASS_MAPPINGS.update(NOISE_PROFILE_CYCLER_CLASS_MAPPINGS)
|
|
NODE_CLASS_MAPPINGS.update(SPEED_SAMPLER_CLASS_MAPPINGS)
|
|
NODE_CLASS_MAPPINGS.update(HIRESFIX_TILED_CLASS_MAPPINGS)
|
|
NODE_CLASS_MAPPINGS.update(TILED_HIRESFIX_ADVANCED_CLASS_MAPPINGS)
|
|
NODE_CLASS_MAPPINGS.update(PIXEL_ANCHORED_REMASTER_CLASS_MAPPINGS)
|
|
NODE_CLASS_MAPPINGS.update(UNET_BLOCK_NOISE_PATCH_CLASS_MAPPINGS)
|
|
NODE_CLASS_MAPPINGS.update(WAN22_VIDEO_CLASS_MAPPINGS)
|
|
|
|
NODE_DISPLAY_NAME_MAPPINGS = {}
|
|
NODE_DISPLAY_NAME_MAPPINGS.update(CHECKPOINT_DISPLAY_MAPPINGS)
|
|
NODE_DISPLAY_NAME_MAPPINGS.update(DIFFUSION_CLIP_VAE_DISPLAY_MAPPINGS)
|
|
NODE_DISPLAY_NAME_MAPPINGS.update(FOUR_PROMPT_MODEL_DISPLAY_MAPPINGS)
|
|
NODE_DISPLAY_NAME_MAPPINGS.update(FOUR_PROMPT_CHECKPOINT_DISPLAY_MAPPINGS)
|
|
NODE_DISPLAY_NAME_MAPPINGS.update(INT_STRING_DISPLAY_MAPPINGS)
|
|
NODE_DISPLAY_NAME_MAPPINGS.update(RANDOM_VOCAB_STRING_LIST_DISPLAY_MAPPINGS)
|
|
NODE_DISPLAY_NAME_MAPPINGS.update(MINIMAX_H3_PROMPT_BUILDER_DISPLAY_MAPPINGS)
|
|
NODE_DISPLAY_NAME_MAPPINGS.update(ACE_SONG_VARIATION_DIRECTOR_DISPLAY_MAPPINGS)
|
|
NODE_DISPLAY_NAME_MAPPINGS.update(ACE_SONG_TIMELINE_DISPLAY_MAPPINGS)
|
|
NODE_DISPLAY_NAME_MAPPINGS.update(AUDIO_TIMELINE_MIXER_DISPLAY_MAPPINGS)
|
|
NODE_DISPLAY_NAME_MAPPINGS.update(OLLAMA_PROMPT_REFINER_DISPLAY_MAPPINGS)
|
|
NODE_DISPLAY_NAME_MAPPINGS.update(OLLAMA_VIDEO_PROMPT_REFINER_DISPLAY_MAPPINGS)
|
|
NODE_DISPLAY_NAME_MAPPINGS.update(OLLAMA_VISION_CAPTIONER_DISPLAY_MAPPINGS)
|
|
NODE_DISPLAY_NAME_MAPPINGS.update(LOAD_IMAGE_DISPLAY_MAPPINGS)
|
|
NODE_DISPLAY_NAME_MAPPINGS.update(T5_DISPLAY_MAPPINGS)
|
|
NODE_DISPLAY_NAME_MAPPINGS.update(T5_EQUAL_LENGTH_DISPLAY_MAPPINGS)
|
|
NODE_DISPLAY_NAME_MAPPINGS.update(T5_SCULPT_EQUAL_LENGTH_DISPLAY_MAPPINGS)
|
|
NODE_DISPLAY_NAME_MAPPINGS.update(CLIP_SCULPT_TEXT_ENCODE_DISPLAY_MAPPINGS)
|
|
NODE_DISPLAY_NAME_MAPPINGS.update(CONDITIONING_VECTOR_OPS_DISPLAY_MAPPINGS)
|
|
NODE_DISPLAY_NAME_MAPPINGS.update(CONDITIONING_TOOLS_DISPLAY_MAPPINGS)
|
|
NODE_DISPLAY_NAME_MAPPINGS.update(REGIONAL_SPLIT_DISPLAY_MAPPINGS)
|
|
NODE_DISPLAY_NAME_MAPPINGS.update(REGIONAL_PROMPT_DISPLAY_MAPPINGS)
|
|
NODE_DISPLAY_NAME_MAPPINGS.update(REGIONAL_SCULPT_PROMPT_DISPLAY_MAPPINGS)
|
|
NODE_DISPLAY_NAME_MAPPINGS.update(WEBSOCKET_DISPLAY_MAPPINGS)
|
|
NODE_DISPLAY_NAME_MAPPINGS.update(ADVANCED_NOISE_SAMPLER_DISPLAY_MAPPINGS)
|
|
NODE_DISPLAY_NAME_MAPPINGS.update(ILLUSTRIOUS_NOISE_SAMPLER_DISPLAY_MAPPINGS)
|
|
NODE_DISPLAY_NAME_MAPPINGS.update(PONY_V7_NOISE_SAMPLER_DISPLAY_MAPPINGS)
|
|
NODE_DISPLAY_NAME_MAPPINGS.update(UNIVERSAL_NOISE_SAMPLER_DISPLAY_MAPPINGS)
|
|
NODE_DISPLAY_NAME_MAPPINGS.update(NOISE_PROFILE_CYCLER_DISPLAY_MAPPINGS)
|
|
NODE_DISPLAY_NAME_MAPPINGS.update(SPEED_SAMPLER_DISPLAY_MAPPINGS)
|
|
NODE_DISPLAY_NAME_MAPPINGS.update(HIRESFIX_TILED_DISPLAY_MAPPINGS)
|
|
NODE_DISPLAY_NAME_MAPPINGS.update(TILED_HIRESFIX_ADVANCED_DISPLAY_MAPPINGS)
|
|
NODE_DISPLAY_NAME_MAPPINGS.update(PIXEL_ANCHORED_REMASTER_DISPLAY_MAPPINGS)
|
|
NODE_DISPLAY_NAME_MAPPINGS.update(UNET_BLOCK_NOISE_PATCH_DISPLAY_MAPPINGS)
|
|
NODE_DISPLAY_NAME_MAPPINGS.update(WAN22_VIDEO_DISPLAY_MAPPINGS)
|
|
|
|
__all__ = ["NODE_CLASS_MAPPINGS", "NODE_DISPLAY_NAME_MAPPINGS", "WEB_DIRECTORY"]
|