第三方插件 ComfyUI-WanVideoWrapper

WanVideo Block Swap

WanVideo Block Swap 261

该节点用于配置 Transformer 块交换(Block Swapping) 策略,以在使用 WanVideo 大模型(如 14B)时减少显存占用。通过将部分 Transformer 块、图像嵌入或文本嵌入在推理过程中转移到 CPU,可以显著节省 VRAM。WanVideo Block

节点中英文对比

按当前节点配置,分别展示中文与英文节点结构。

中文节点
WanVideo Block Swap
  • block_swap_args
  • blocks_to_swap
    offload_img_emb
    offload_txt_emb
    use_non_blocking
    vace_blocks_to_swap
    英文节点
    WanVideo Block Swap
  • block_swap_args
  • blocks_to_swap
    offload_img_emb
    offload_txt_emb
    use_non_blocking
    vace_blocks_to_swap

    参数说明

    依据当前节点关联的 `NodesItems` 数据展示输入、输出与控件说明。

    输出参数
    block_swap_args block_swap_args
    包含卸载配置的字典结构,供 Wan 模型加载节点使用。
    控件参数
    blocks_to_swap blocks_to_swap
    需要被卸载(swap)的 Transformer 层数,14B 模型有 40 层,1.3B 模型有 30 层。建议设置为 10–25,根据显存情况决定;值越大节省显存越多但访问延迟越高。
    offload_img_emb offload_img_emb
    是否卸载图像 embedding 模块(img_emb)到 CPU。
    offload_txt_emb offload_txt_emb
    是否卸载文本 embedding 模块(txt_emb)到 CPU。
    use_non_blocking use_non_blocking
    是否启用非阻塞内存传输,提高 swap 速度但会多占用系统内存。建议在内存充足的系统上启用此项,能显著提升推理响应时间。
    vace_blocks_to_swap vace_blocks_to_swap
    若使用 VACE 模型,设置需要 swap 的 VACE 层数(最多支持 15 层)。建议在使用 t2v 模型(含 vace_blocks)时设置为 6–10,默认 0 表示不卸载。

    详细说明

    结合节点用途、参数和调用方式,给出更完整的说明内容。

    暂无节点说明

    同类节点导航

    继续浏览同一分类下的相邻节点,方便串联学习。