400 lines
23 KiB
Python
400 lines
23 KiB
Python
import os
|
|
import traceback
|
|
|
|
from server import PromptServer
|
|
from aiohttp import web
|
|
|
|
HAS_TKINTER = False
|
|
try:
|
|
import tkinter as tk
|
|
from tkinter import filedialog
|
|
HAS_TKINTER = True
|
|
except ImportError:
|
|
pass # 静默放行，不要让整个节点加载失败
|
|
|
|
@PromptServer.instance.routes.post("/xb_toolbox/choose_folder")
|
|
async def choose_folder(request):
|
|
if not HAS_TKINTER:
|
|
return web.json_response({"path": "", "error": "当前整合包环境缺少弹窗依赖，请手动输入或粘贴文件夹路径。"})
|
|
|
|
try:
|
|
root = tk.Tk()
|
|
root.withdraw()
|
|
root.attributes('-topmost', True)
|
|
folder_path = filedialog.askdirectory()
|
|
root.destroy()
|
|
return web.json_response({"path": folder_path})
|
|
except Exception as e:
|
|
return web.json_response({"path": "", "error": f"弹窗调用失败: {str(e)}"})
|
|
|
|
|
|
from .nodes_audio_slicer import handle_audio_waveform
|
|
|
|
@PromptServer.instance.routes.post("/xb_toolbox/audio_waveform")
|
|
async def get_audio_waveform(request):
|
|
return await handle_audio_waveform(request)
|
|
|
|
|
|
def print_success(msg):
|
|
print(f"\033[92m{msg}\033[0m")
|
|
|
|
def print_error(msg):
|
|
print(f"\033[91m{msg}\033[0m")
|
|
|
|
def print_warning(msg):
|
|
print(f"\033[93m{msg}\033[0m")
|
|
|
|
NODE_CLASS_MAPPINGS = {}
|
|
NODE_DISPLAY_NAME_MAPPINGS = {}
|
|
WEB_DIRECTORY = os.path.join(os.path.dirname(os.path.abspath(__file__)), "js")
|
|
|
|
try:
|
|
from .nodes_vis import XB_VRAM_Calculator, XB_ChunkVisualization
|
|
from .nodes_vram import XTX_Data_Radar
|
|
from .nodes_video import XB_VideoParamsMaster, XB_ImageParamsMaster, XB_MasterParameter, XB_VideoLoader, XB_VideoCombine
|
|
from .nodes_blockswap import XB_UNetBlockSwap, XB_CheckpointBlockSwap
|
|
from .nodes_blockswap_sage import XB_Sage_BlockSwap
|
|
from .nodes_wiring import XB_DynamicBus, XB_UNetNameBroadcaster, XB_CLIPNameBroadcaster
|
|
from .nodes_dashboard import XB_Dashboard_Zen
|
|
from .nodes_memory_viz import XB_MemoryVisualization
|
|
from .nodes_tile import XB_SamplerChunkMaster
|
|
from .nodes_wan_vae import (XB_WanImageToVideo, XB_WanFirstLastFrameToVideo, XB_WanSoundImageToVideo, XB_WanFunControlToVideo, XB_WanVaceToVideo, XB_Wan22FunControlToVideo, XB_WanInfiniteTalkToVideo, XB_WanInfiniteTalkToVideo_Single, XB_WanInfiniteTalkToVideo_Dual, XB_WanVAEDecodeTiled, XB_WanFunInpaintToVideo, XB_WanCameraImageToVideo, XB_WanPhantomSubjectToVideo, XB_WanHuMoImageToVideo, XB_Wan22ImageToVideoLatent, XB_WanSoundImageToVideoExtend, XB_WanSCAILToVideo, XB_WanSCAILToVideoPro, XB_WanDancerVideo, XB_WanDanceSwitcher, XB_WanDancerCombo, XB_BerniniConditioning)
|
|
from .nodes_batch import XB_BatchFolderLoader
|
|
from .nodes_pipeline import XB_Wan_ParamBus, XB_Wan_RelayNode, XB_Wan_InfiniteRelayNode, XB_Video_Merger, XB_StoryboardSlicer,XB_WanAnimate_ParamBus,XB_WanAnimate_RelayNode, XB_WanInfiniteTalk_ParamBus, XB_WanInfiniteTalk_RelayNode, XB_Wan_InfiniteRelayNode_New, XB_WanAnimate_RelayNode_New, XB_WanInfiniteTalk_RelayNode_New, XB_WanSCAIL_ParamBus_New, XB_WanSCAIL_RelayNode_New, XB_WanInfiniteTalk_RelayNode_MultiRef, XB_WanInfiniteTalk_RelayNode_AllInOne
|
|
from .nodes_sageatt import XB_SageAttentionAccelerator
|
|
from .nodes_wan_t5 import XB_WanT5Loader
|
|
from .nodes_wan import (XB_WanCompileSettings, XB_WanModelLoader, XB_WanBlockSwap,
|
|
XB_WanSampler, XB_WanTextEncode, XB_WanVAELoader, XB_WanDecode,
|
|
XB_WanAnimateToVideo)
|
|
|
|
from .nodes_label import XB_CanvasLabel
|
|
from .nodes_audio_slicer import XB_AudioSlicer, XB_AudioSlicerV1, XB_AudioSlicerV2, XB_AudioSlicerV3
|
|
from .nodes_segmentation import XB_HumanSegModelLoader, XB_HumanSegmentation
|
|
|
|
# 注册视频转码端点（预览窗口实时缩放依赖）
|
|
from . import xb_video_server
|
|
from .nodes_rocm import XB_ROCmMemCleaner
|
|
from .nodes_vanilla_wrappers import (XB_KSampler, XB_KSamplerAdvanced,
|
|
XB_SamplerCustom, XB_SamplerCustomAdvanced,
|
|
XB_VAEDecode, XB_VAEDecodeTiled,
|
|
XB_VAEDecodeTiledImage,
|
|
XB_VAEEncode, XB_VAEEncodeTiled,
|
|
XB_VAEEncodeForInpaint,
|
|
_AliasKSampler, _AliasKSamplerAdvanced,
|
|
_AliasSamplerCustom, _AliasSamplerCustomAdvanced,
|
|
_AliasVAEDecode, _AliasVAEDecodeTemporal,
|
|
_AliasLTXVAEDecode, _AliasVAEEncode)
|
|
from .nodes_string_merge import XB_StringMerge
|
|
from .nodes_msr import XB_MSR
|
|
from .nodes_comic import XB_ComicPromptParser, XB_ComicTextRenderer, XB_AutoBubbleTextRenderer
|
|
from .nodes_super import XB_BerniniPromptRouter, XB_K2StyleRouter
|
|
from .nodes_int8_rocm import (
|
|
UNetLoaderINTROCm, INT8GroupedLoraROCm, INT8LoraROCm, INT8ModelSaveROCm, PreLoraLoaderROCm
|
|
)
|
|
from .nodes_int8_clip_rocm import (
|
|
CLIPLoaderINT8ROCm, DualCLIPLoaderINT8ROCm, INT8CLIPSaveROCm
|
|
)
|
|
|
|
# ── Llama 节点 (LLM/VLM) ──
|
|
try:
|
|
from .nodes_llama import (
|
|
XB_llamaModelLoader, XB_llamaInstruct, XB_llamaParameters,
|
|
XB_llamaUnloadModel, XB_llamaCleanStates, XB_llamaParseJSON,
|
|
XB_llamaJSON2BBox, XB_llamaBBox2SEGS, XB_llamaBBox2Mask,
|
|
XB_llamaBBoxes2BBox, XB_llamaUnpackCodeBlock, XB_llamaPromptEnhancer,
|
|
)
|
|
_LLAMA_AVAILABLE = True
|
|
except ImportError as e:
|
|
print_warning(f"[XB-llama] llama-cpp-python 未安装, Llama 节点不可用: {e}")
|
|
_LLAMA_AVAILABLE = False
|
|
|
|
# ── CosyVoice3 音频节点 ──
|
|
from .cosyvoice3.nodes.model_loader import XB_CosyVoice3_ModelLoader
|
|
from .cosyvoice3.nodes.zero_shot import XB_CosyVoice3_ZeroShot
|
|
from .cosyvoice3.nodes.cross_lingual import XB_CosyVoice3_CrossLingual
|
|
from .cosyvoice3.nodes.voice_conversion import XB_CosyVoice3_VoiceConversion
|
|
from .cosyvoice3.nodes.audio_crop import XB_CosyVoice3_AudioCrop
|
|
from .cosyvoice3.nodes.dialog import XB_CosyVoice3_Dialog
|
|
from .cosyvoice3.nodes.instruct2 import XB_CosyVoice3_Instruct2
|
|
from .cosyvoice3.nodes.save_speaker import XB_CosyVoice3_SaveSpeaker
|
|
from .cosyvoice3.nodes.speaker_clone import XB_CosyVoice3_SpeakerClone
|
|
from .cosyvoice3.nodes.speaker_instruct2 import XB_CosyVoice3_SpeakerInstruct2
|
|
|
|
NODE_CLASS_MAPPINGS = {
|
|
"XB_VRAM_Calculator": XB_VRAM_Calculator,
|
|
"XB_ChunkVisualization": XB_ChunkVisualization,
|
|
"XTX_Data_Radar": XTX_Data_Radar,
|
|
"XB_VideoParamsMaster": XB_VideoParamsMaster,
|
|
"XB_ImageParamsMaster": XB_ImageParamsMaster,
|
|
"XB_MasterParameter": XB_MasterParameter,
|
|
"XB_VideoLoader": XB_VideoLoader,
|
|
"XB_VideoCombine": XB_VideoCombine,
|
|
"XB_UNetBlockSwap": XB_UNetBlockSwap,
|
|
"XB_CheckpointBlockSwap": XB_CheckpointBlockSwap,
|
|
"XB_Sage_BlockSwap": XB_Sage_BlockSwap,
|
|
"XB_DynamicBus": XB_DynamicBus,
|
|
"XB_UNetNameBroadcaster": XB_UNetNameBroadcaster,
|
|
"XB_CLIPNameBroadcaster": XB_CLIPNameBroadcaster,
|
|
"XB_Dashboard_Zen": XB_Dashboard_Zen,
|
|
"XB_MemoryVisualization": XB_MemoryVisualization,
|
|
"XB_SamplerChunkMaster": XB_SamplerChunkMaster,
|
|
"XB_WanImageToVideo": XB_WanImageToVideo,
|
|
"XB_WanFirstLastFrameToVideo": XB_WanFirstLastFrameToVideo,
|
|
"XB_WanFunControlToVideo": XB_WanFunControlToVideo,
|
|
"XB_WanVaceToVideo": XB_WanVaceToVideo,
|
|
"XB_Wan22FunControlToVideo": XB_Wan22FunControlToVideo,
|
|
"XB_WanSoundImageToVideo": XB_WanSoundImageToVideo,
|
|
"XB_WanInfiniteTalkToVideo": XB_WanInfiniteTalkToVideo,
|
|
"XB_WanInfiniteTalkToVideo_Single": XB_WanInfiniteTalkToVideo_Single,
|
|
"XB_WanInfiniteTalkToVideo_Dual": XB_WanInfiniteTalkToVideo_Dual,
|
|
"XB_WanVAEDecodeTiled": XB_WanVAEDecodeTiled,
|
|
"XB_WanFunInpaintToVideo": XB_WanFunInpaintToVideo,
|
|
"XB_WanCameraImageToVideo": XB_WanCameraImageToVideo,
|
|
"XB_WanPhantomSubjectToVideo": XB_WanPhantomSubjectToVideo,
|
|
"XB_WanHuMoImageToVideo": XB_WanHuMoImageToVideo,
|
|
"XB_Wan22ImageToVideoLatent": XB_Wan22ImageToVideoLatent,
|
|
"XB_WanSoundImageToVideoExtend": XB_WanSoundImageToVideoExtend,
|
|
"XB_WanSCAILToVideo": XB_WanSCAILToVideo,
|
|
"XB_WanSCAILToVideoPro": XB_WanSCAILToVideoPro,
|
|
"XB_WanDancerVideo": XB_WanDancerVideo,
|
|
"XB_WanDanceSwitcher": XB_WanDanceSwitcher,
|
|
"XB_WanDancerCombo": XB_WanDancerCombo,
|
|
"XB_BerniniConditioning": XB_BerniniConditioning,
|
|
"XB_BatchFolderLoader": XB_BatchFolderLoader,
|
|
"XB_Wan_ParamBus": XB_Wan_ParamBus,
|
|
"XB_Wan_RelayNode": XB_Wan_RelayNode,
|
|
"XB_Wan_InfiniteRelayNode": XB_Wan_InfiniteRelayNode,
|
|
"XB_Video_Merger": XB_Video_Merger,
|
|
"XB_StoryboardSlicer": XB_StoryboardSlicer,
|
|
"XB_SageAttentionAccelerator": XB_SageAttentionAccelerator,
|
|
"XB_ROCmMemCleaner": XB_ROCmMemCleaner,
|
|
# ── 借尸还魂：旧 ROCm 节点 ID → 别名类（兼容旧参数名） ──
|
|
"XB_ROCmKSampler": _AliasKSampler,
|
|
"XB_ROCmKSamplerAdvanced": _AliasKSamplerAdvanced,
|
|
"XB_ROCmSamplerCustom": _AliasSamplerCustom,
|
|
"XB_ROCmSamplerCustomAdvanced": _AliasSamplerCustomAdvanced,
|
|
"XB_ROCmVAEDecode": _AliasVAEDecode,
|
|
"XB_ROCmVAEEncode": _AliasVAEEncode,
|
|
"XB_ROCmVAEDecodeTemporal": _AliasVAEDecodeTemporal,
|
|
"XB_ROCmLTXVAEDecode": _AliasLTXVAEDecode,
|
|
# ── 新原版优化节点 ──
|
|
"XB_KSampler": XB_KSampler,
|
|
"XB_KSamplerAdvanced": XB_KSamplerAdvanced,
|
|
"XB_SamplerCustom": XB_SamplerCustom,
|
|
"XB_SamplerCustomAdvanced": XB_SamplerCustomAdvanced,
|
|
"XB_VAEDecode": XB_VAEDecode,
|
|
"XB_VAEDecodeTiled": XB_VAEDecodeTiled,
|
|
"XB_VAEDecodeTiledImage": XB_VAEDecodeTiledImage,
|
|
"XB_VAEEncode": XB_VAEEncode,
|
|
"XB_VAEEncodeTiled": XB_VAEEncodeTiled,
|
|
"XB_VAEEncodeForInpaint": XB_VAEEncodeForInpaint,
|
|
"XB_WanT5Loader": XB_WanT5Loader,
|
|
"XB_WanCompileSettings": XB_WanCompileSettings,
|
|
"XB_WanModelLoader": XB_WanModelLoader,
|
|
"XB_WanBlockSwap": XB_WanBlockSwap,
|
|
"XB_WanSampler": XB_WanSampler,
|
|
"XB_WanTextEncode": XB_WanTextEncode,
|
|
"XB_WanVAELoader": XB_WanVAELoader,
|
|
"XB_WanDecode": XB_WanDecode,
|
|
"XB_WanAnimateToVideo": XB_WanAnimateToVideo,
|
|
"XB_WanAnimate_ParamBus": XB_WanAnimate_ParamBus,
|
|
"XB_WanAnimate_RelayNode": XB_WanAnimate_RelayNode,
|
|
"XB_WanInfiniteTalk_ParamBus": XB_WanInfiniteTalk_ParamBus,
|
|
"XB_WanInfiniteTalk_RelayNode": XB_WanInfiniteTalk_RelayNode,
|
|
"XB_Wan_InfiniteRelayNode_New": XB_Wan_InfiniteRelayNode_New,
|
|
"XB_WanAnimate_RelayNode_New": XB_WanAnimate_RelayNode_New,
|
|
"XB_WanInfiniteTalk_RelayNode_New": XB_WanInfiniteTalk_RelayNode_New,
|
|
"XB_WanSCAIL_ParamBus_New": XB_WanSCAIL_ParamBus_New,
|
|
"XB_WanSCAIL_RelayNode_New": XB_WanSCAIL_RelayNode_New,
|
|
"XB_WanInfiniteTalk_RelayNode_MultiRef": XB_WanInfiniteTalk_RelayNode_MultiRef,
|
|
"XB_WanInfiniteTalk_RelayNode_AllInOne": XB_WanInfiniteTalk_RelayNode_AllInOne,
|
|
"XB_HumanSegmentation": XB_HumanSegmentation,
|
|
"XB_HumanSegModelLoader": XB_HumanSegModelLoader,
|
|
"XB_CanvasLabel": XB_CanvasLabel,
|
|
"XB_AudioSlicer": XB_AudioSlicer,
|
|
"XB_AudioSlicerV1": XB_AudioSlicerV1,
|
|
"XB_AudioSlicerV2": XB_AudioSlicerV2,
|
|
"XB_AudioSlicerV3": XB_AudioSlicerV3,
|
|
"XB_StringMerge": XB_StringMerge,
|
|
"XB_MSR": XB_MSR,
|
|
"XB_ComicPromptParser": XB_ComicPromptParser,
|
|
"XB_ComicTextRenderer": XB_ComicTextRenderer,
|
|
"XB_AutoBubbleTextRenderer": XB_AutoBubbleTextRenderer,
|
|
"XB_BerniniPromptRouter": XB_BerniniPromptRouter,
|
|
"XB_K2StyleRouter": XB_K2StyleRouter,
|
|
# ── INT8 ROCm 量化节点 ──
|
|
"XB_UNetLoaderINTROCm": UNetLoaderINTROCm,
|
|
"XB_INT8LoraROCm": INT8LoraROCm,
|
|
"XB_INT8GroupedLoraROCm": INT8GroupedLoraROCm,
|
|
"XB_INT8ModelSaveROCm": INT8ModelSaveROCm,
|
|
"XB_INT8PreLoraLoaderROCm": PreLoraLoaderROCm,
|
|
"XB_CLIPLoaderINT8ROCm": CLIPLoaderINT8ROCm,
|
|
"XB_DualCLIPLoaderINT8ROCm": DualCLIPLoaderINT8ROCm,
|
|
"XB_INT8CLIPSaveROCm": INT8CLIPSaveROCm,
|
|
# ── CosyVoice3 音频节点 ──
|
|
"XB_CosyVoice3_ModelLoader": XB_CosyVoice3_ModelLoader,
|
|
"XB_CosyVoice3_ZeroShot": XB_CosyVoice3_ZeroShot,
|
|
"XB_CosyVoice3_CrossLingual": XB_CosyVoice3_CrossLingual,
|
|
"XB_CosyVoice3_VoiceConversion": XB_CosyVoice3_VoiceConversion,
|
|
"XB_CosyVoice3_AudioCrop": XB_CosyVoice3_AudioCrop,
|
|
"XB_CosyVoice3_Dialog": XB_CosyVoice3_Dialog,
|
|
"XB_CosyVoice3_Instruct2": XB_CosyVoice3_Instruct2,
|
|
"XB_CosyVoice3_SaveSpeaker": XB_CosyVoice3_SaveSpeaker,
|
|
"XB_CosyVoice3_SpeakerClone": XB_CosyVoice3_SpeakerClone,
|
|
"XB_CosyVoice3_SpeakerInstruct2": XB_CosyVoice3_SpeakerInstruct2,
|
|
}
|
|
|
|
# ── 条件注册 Llama 节点 ──
|
|
if _LLAMA_AVAILABLE:
|
|
NODE_CLASS_MAPPINGS.update({
|
|
"XB_llamaModelLoader": XB_llamaModelLoader,
|
|
"XB_llamaInstruct": XB_llamaInstruct,
|
|
"XB_llamaParameters": XB_llamaParameters,
|
|
"XB_llamaUnloadModel": XB_llamaUnloadModel,
|
|
"XB_llamaCleanStates": XB_llamaCleanStates,
|
|
"XB_llamaParseJSON": XB_llamaParseJSON,
|
|
"XB_llamaJSON2BBox": XB_llamaJSON2BBox,
|
|
"XB_llamaBBox2SEGS": XB_llamaBBox2SEGS,
|
|
"XB_llamaBBox2Mask": XB_llamaBBox2Mask,
|
|
"XB_llamaBBoxes2BBox": XB_llamaBBoxes2BBox,
|
|
"XB_llamaUnpackCodeBlock": XB_llamaUnpackCodeBlock,
|
|
"XB_llamaPromptEnhancer": XB_llamaPromptEnhancer,
|
|
})
|
|
|
|
NODE_DISPLAY_NAME_MAPPINGS = {
|
|
"XB_VRAM_Calculator": "XB-BOX - VRAM Calculator",
|
|
"XB_ChunkVisualization": "XB-BOX - Chunk Visualization",
|
|
"XTX_Data_Radar": "XB-BOX - Data Radar",
|
|
"XB_VideoParamsMaster": "XB-BOX - Video Params Master",
|
|
"XB_ImageParamsMaster": "XB-BOX - Image Params Master",
|
|
"XB_MasterParameter": "XB-BOX - Master Parameter",
|
|
"XB_VideoLoader": "XB-BOX - 🎬 视频加载器（修复预览BUG）",
|
|
"XB_VideoCombine": "XB-BOX - 🎬 视频拼接输出",
|
|
"XB_UNetBlockSwap": "XB-BOX - UNet Block Swap",
|
|
"XB_CheckpointBlockSwap": "XB-BOX - Checkpoint Block Swap",
|
|
"XB_Sage_BlockSwap": "XB-BOX - Sage + BlockSwap (Golden Duo)",
|
|
"XB_DynamicBus": "XB-BOX - Dynamic Bus",
|
|
"XB_UNetNameBroadcaster": "XB-BOX - UNet Name Broadcaster",
|
|
"XB_CLIPNameBroadcaster": "XB-BOX - CLIP Name Broadcaster",
|
|
"XB_Dashboard_Zen": "XB-BOX - Dashboard Zen",
|
|
"XB_MemoryVisualization": "XB 📊 硬件监控面板 (Memory Visualization)",
|
|
"XB_SamplerChunkMaster": "XB-BOX - Sampler Chunk Master",
|
|
"XB_WanImageToVideo": "XB-BOX - Wan Image2Video",
|
|
"XB_WanFirstLastFrameToVideo": "XB-BOX - Wan First/Last Frame2Video",
|
|
"XB_BatchFolderLoader": "XB-BOX - Batch Folder Loader",
|
|
"XB_Wan_ParamBus": "XB-BOX - Wan Param Bus",
|
|
"XB_Wan_RelayNode": "XB-BOX - Wan Relay Node",
|
|
"XB_Wan_InfiniteRelayNode": "XB-BOX - Wan Infinite Relay Node",
|
|
"XB_Video_Merger": "XB-BOX - Video Merger",
|
|
"XB_StoryboardSlicer": "XB-BOX - Storyboard Slicer",
|
|
"XB_SageAttentionAccelerator": "XB-BOX - SageAttention Accelerator",
|
|
"XB_ROCmMemCleaner": "XB-BOX - 🧹 显存清理",
|
|
# ── 借尸还魂：旧节点名 + 新的优化版显示名 ──
|
|
"XB_ROCmKSampler": "XB-BOX - 采样器（原版优化）",
|
|
"XB_ROCmKSamplerAdvanced": "XB-BOX - 高级采样器（原版优化）",
|
|
"XB_ROCmSamplerCustom": "XB-BOX - 自定义采样器（原版优化）",
|
|
"XB_ROCmSamplerCustomAdvanced": "XB-BOX - 自定义高级采样器（原版优化）",
|
|
"XB_ROCmVAEDecode": "XB-BOX - VAE解码（原版优化）",
|
|
"XB_ROCmVAEEncode": "XB-BOX - VAE编码（原版优化）",
|
|
"XB_ROCmVAEDecodeTemporal": "XB-BOX - VAE分块解码（原版优化）",
|
|
"XB_ROCmLTXVAEDecode": "XB-BOX - VAE分块解码（原版优化）",
|
|
# ── 新优化版节点 ──
|
|
"XB_KSampler": "XB-BOX - 采样器（原版优化）",
|
|
"XB_KSamplerAdvanced": "XB-BOX - 高级采样器（原版优化）",
|
|
"XB_SamplerCustom": "XB-BOX - 自定义采样器（原版优化）",
|
|
"XB_SamplerCustomAdvanced": "XB-BOX - 自定义高级采样器（原版优化）",
|
|
"XB_VAEDecode": "XB-BOX - VAE解码（原版优化）",
|
|
"XB_VAEDecodeTiled": "XB-BOX - VAE分块解码（原版优化）",
|
|
"XB_VAEDecodeTiledImage": "XB-BOX - VAE解码（原版优化）",
|
|
"XB_VAEEncode": "XB-BOX - VAE编码（原版优化）",
|
|
"XB_VAEEncodeTiled": "XB-BOX - VAE分块编码（原版优化）",
|
|
"XB_VAEEncodeForInpaint": "XB-BOX - VAE修补编码（原版优化）",
|
|
"XB_WanT5Loader": "XB-BOX - 📝 Wan T5 加载器(FP8)",
|
|
"XB_WanCompileSettings": "XB-BOX - ⚡ Wan 编译设置",
|
|
"XB_WanModelLoader": "XB-BOX - 🧠 Wan 模型加载",
|
|
"XB_WanBlockSwap": "XB-BOX - 🔄 Wan 分块交换",
|
|
"XB_WanSampler": "XB-BOX - 🎯 Wan 采样器",
|
|
"XB_WanTextEncode": "XB-BOX - ✍️ Wan 文本编码",
|
|
"XB_WanVAELoader": "XB-BOX - 🎨 Wan VAE 加载",
|
|
"XB_WanDecode": "XB-BOX - 🖼️ Wan VAE 解码",
|
|
"XB_WanAnimateToVideo": "XB-BOX - 🎬 Wan 动画转视频",
|
|
"XB_WanAnimate_ParamBus": "XB-BOX - 🎬 Animate 动作迁移总线",
|
|
"XB_WanAnimate_RelayNode": "XB-BOX - 🏃♀️ Animate 无限接力点",
|
|
"XB_WanInfiniteTalk_ParamBus": "XB-BOX - 🎵 InfiniteTalk 无限对口型总线",
|
|
"XB_WanInfiniteTalk_RelayNode": "XB-BOX - 🏃 InfiniteTalk 无限对口型接力点",
|
|
"XB_Wan_InfiniteRelayNode_New": "XB-BOX - 🆕 Wan 无限接力点 (New)",
|
|
"XB_WanAnimate_RelayNode_New": "XB-BOX - 🆕 Animate 无限接力点 (New)",
|
|
"XB_WanInfiniteTalk_RelayNode_New": "XB-BOX - 🆕 InfiniteTalk 无限接力点 (New)",
|
|
"XB_WanSCAIL_ParamBus_New": "XB-BOX - 🆕 SCAIL 总线 (New)",
|
|
"XB_WanSCAIL_RelayNode_New": "XB-BOX - 🆕 SCAIL 无限接力点 (New)",
|
|
"XB_WanInfiniteTalk_RelayNode_MultiRef": "XB-BOX - 🆕 InfiniteTalk 无限接力点 (多图)",
|
|
"XB_WanInfiniteTalk_RelayNode_AllInOne": "XB-BOX - 🆕 InfiniteTalk 无限接力点 (全能)",
|
|
"XB_HumanSegmentation": "XB-BOX - ✂️ 人物分割 (DirectML/ROCm)",
|
|
"XB_HumanSegModelLoader": "XB-BOX - 📥 人物分割模型加载",
|
|
"XB_CanvasLabel": "XB-BOX - 🏷️ Canvas Label (文字标签)",
|
|
"XB_AudioSlicer": "XB-BOX - 🎵 音频切片（基础）",
|
|
"XB_AudioSlicerV1": "XB-BOX - 🎵 音频切片V1（单人）",
|
|
"XB_AudioSlicerV2": "XB-BOX - 🎵 音频切片V2（双人）",
|
|
"XB_AudioSlicerV3": "XB-BOX - 🎵 音频切片V3（高级）",
|
|
"XB_WanInfiniteTalkToVideo": "XB-BOX - 🎵 语音转视频分块",
|
|
"XB_WanInfiniteTalkToVideo_Single": "XB-BOX - 🎵 语音转视频分块（单人）",
|
|
"XB_WanInfiniteTalkToVideo_Dual": "XB-BOX - 🎵 语音转视频分块（双人）",
|
|
"XB_StringMerge": "XB-BOX - 📝 字符串合并",
|
|
"XB_MSR": "XB-BOX - 🎞️ MSR 多图合成帧序列",
|
|
"XB_ComicPromptParser": "XB-BOX - 📝 漫画提示词智能解析",
|
|
"XB_ComicTextRenderer": "XB-BOX - 💬 漫画文字渲染 (精确坐标)",
|
|
"XB_AutoBubbleTextRenderer": "XB-BOX - 🤖 漫画文字渲染 (全自动带涂改液)",
|
|
"XB_BerniniConditioning": "XB-BOX - 🎨 Bernini 条件注入（VAE分块）",
|
|
"XB_BerniniPromptRouter": "XB-BOX - 🧠 Bernini 提示词路由（超级节点）",
|
|
"XB_K2StyleRouter": "XB-BOX - 🎨 K2 风格切换路由（超级节点）",
|
|
# ── INT8 ROCm 量化节点 ──
|
|
"XB_UNetLoaderINTROCm": "XB-BOX - 🧊 INT8 UNet 加载器 (ROCm)",
|
|
"XB_INT8LoraROCm": "XB-BOX - 🧊 INT8 单LoRA (ROCm)",
|
|
"XB_INT8GroupedLoraROCm": "XB-BOX - 🧊 INT8 组合 LoRA (ROCm)",
|
|
"XB_INT8ModelSaveROCm": "XB-BOX - 🧊 INT8 模型保存 (ROCm)",
|
|
"XB_INT8PreLoraLoaderROCm": "XB-BOX - 🧊 INT8 预加载 LoRA (ROCm)",
|
|
"XB_CLIPLoaderINT8ROCm": "XB-BOX - 🧊 INT8 CLIP 加载器 (ROCm)",
|
|
"XB_DualCLIPLoaderINT8ROCm": "XB-BOX - 🧊 INT8 双CLIP加载器 (ROCm)",
|
|
"XB_INT8CLIPSaveROCm": "XB-BOX - 🧊 INT8 CLIP 保存 (ROCm)",
|
|
# ── CosyVoice3 音频节点 ──
|
|
"XB_CosyVoice3_ModelLoader": "XB-BOX - 🔊 CosyVoice3 模型加载",
|
|
"XB_CosyVoice3_ZeroShot": "XB-BOX - 🎤 CosyVoice3 零样本克隆",
|
|
"XB_CosyVoice3_CrossLingual": "XB-BOX - 🌐 CosyVoice3 跨语言合成",
|
|
"XB_CosyVoice3_VoiceConversion": "XB-BOX - 🔄 CosyVoice3 语音转换",
|
|
"XB_CosyVoice3_AudioCrop": "XB-BOX - ✂️ CosyVoice3 音频裁剪",
|
|
"XB_CosyVoice3_Dialog": "XB-BOX - 💬 CosyVoice3 多人对话",
|
|
"XB_CosyVoice3_Instruct2": "XB-BOX - 🎭 CosyVoice3 指令控制",
|
|
"XB_CosyVoice3_SaveSpeaker": "XB-BOX - 💾 CosyVoice3 保存说话人",
|
|
"XB_CosyVoice3_SpeakerClone": "XB-BOX - 🗣️ CosyVoice3 说话人克隆",
|
|
"XB_CosyVoice3_SpeakerInstruct2": "XB-BOX - 🎭 CosyVoice3 说话人指令",
|
|
}
|
|
|
|
# ── 条件注册 Llama 显示名 ──
|
|
if _LLAMA_AVAILABLE:
|
|
NODE_DISPLAY_NAME_MAPPINGS.update({
|
|
"XB_llamaModelLoader": "XB-llama - 📦 模型加载器",
|
|
"XB_llamaInstruct": "XB-llama - 💬 指令推理",
|
|
"XB_llamaParameters": "XB-llama - ⚙️ 推理参数",
|
|
"XB_llamaUnloadModel": "XB-llama - 🗑️ 卸载模型",
|
|
"XB_llamaCleanStates": "XB-llama - 🧹 清理状态",
|
|
"XB_llamaParseJSON": "XB-llama - 📋 解析JSON",
|
|
"XB_llamaJSON2BBox": "XB-llama - 🎯 JSON转BBox",
|
|
"XB_llamaBBox2SEGS": "XB-llama - ✂️ BBox转SEGS",
|
|
"XB_llamaBBox2Mask": "XB-llama - 🎭 BBox转Mask",
|
|
"XB_llamaBBoxes2BBox": "XB-llama - 🔍 BBoxes取BBox",
|
|
"XB_llamaUnpackCodeBlock": "XB-llama - 📝 解包代码块",
|
|
"XB_llamaPromptEnhancer": "XB-llama - ✨ 提示词增强预设",
|
|
})
|
|
|
|
print_success("\n" + "="*50)
|
|
print_success("🚀 [XB-BOX] XB_ToolBox Core Modules Loaded Successfully!")
|
|
print_success("="*50 + "\n")
|
|
|
|
except Exception as e:
|
|
print_error("\n" + "="*50)
|
|
print_error("🚨 [XB-BOX] FATAL ERROR: XB_ToolBox Loading Failed!")
|
|
print_error(f"❌ Error Detail: {str(e)}")
|
|
traceback.print_exc()
|
|
print_error("="*50 + "\n")
|
|
|
|
__all__ = ['NODE_CLASS_MAPPINGS', 'NODE_DISPLAY_NAME_MAPPINGS', 'WEB_DIRECTORY'] |