ComfyUI-FlashVSR_Ultra_Fast

在低显存下运行 FlashVSR，无任何伪影。 [📃中文版本]

更新日志

2025-10-24

• 新增长视频处理流程，在提升长视频分辨率时显著降低显存占用。

2025-10-21

• 项目初始化，引入 tile_dit（分块 DiT）等功能，显著降低显存占用。

2025-10-22

• 用 Sparse_Sage 替换 Block-Sparse-Attention（块稀疏注意力），无需编译任何自定义内核（custom kernels）。
• 新增对 RTX 50 系列 GPU 的支持。

预览

使用说明

• mode（模式）： tiny -> 速度更快（默认）；full -> 质量更高
• scale（缩放比例）： 4 效果始终更好，除非显存不足则使用 2
• color_fix（色彩校正）： 使用小波变换（wavelet transform）校正输出视频的色彩。
• tiled_vae（瓦片 VAE）： 设为 True 可在解码时降低显存占用，但会牺牲速度。
• tiled_dit（瓦片 DiT）： 显著降低显存占用，但会牺牲速度。
• tile_size, tile_overlap（块大小、块重叠）： 如何分割输入视频。
• unload_dit（卸载 DiT）： 在解码前卸载 DiT 以降低显存峰值，但会牺牲速度。

安装

节点：

cd ComfyUI/custom_nodes
git clone https://github.com/lihaoyun6/ComfyUI-FlashVSR_Ultra_Fast.git
python -m pip install -r ComfyUI-FlashVSR_Ultra_Fast/requirements.txt

📢: 对于 Turing 或更早的 GPU，请安装 triton<3.3.0：

# Windows
python -m pip install -U triton-windows<3.3.0
# Linux
python -m pip install -U triton<3.3.0

模型：

• 从这里下载完整的 FlashVSR 文件夹及其所有文件，并放入 ComfyUI/models

├── ComfyUI/models/FlashVSR
|     ├── LQ_proj_in.ckpt
|     ├── TCDecoder.ckpt
|     ├── diffusion_pytorch_model_streaming_dmd.safetensors
|     ├── Wan2.1_VAE.pth

致谢

• FlashVSR @OpenImagingLab
• Sparse_SageAttention @jt-zhang
• ComfyUI @comfyanonymous

ComfyUI-FlashVSR-Ultra-Fast 快速上手指南

环境准备

• 操作系统：Windows 或 Linux
• 硬件要求：NVIDIA GPU（支持 RTX 50 系列）
• 前置软件：
  • ComfyUI 已安装
  • Python 3.x
  • Git

📢 针对 Turing 或更早的 GPU，需安装 triton<3.3.0

安装步骤

1. 克隆节点仓库

cd ComfyUI/custom_nodes
git clone https://github.com/lihaoyun6/ComfyUI-FlashVSR_Ultra_Fast.git

2. 安装依赖

cd ComfyUI/custom_nodes/ComfyUI-FlashVSR_Ultra_Fast
python -m pip install -r requirements.txt

旧显卡额外安装（Turing 或更早 GPU）：

# Windows
python -m pip install -U triton-windows<3.3.0

# Linux
python -m pip install -U triton<3.3.0

3. 下载模型

从 HuggingFace 下载 FlashVSR 文件夹完整内容，放入 ComfyUI/models 目录：

ComfyUI/models/FlashVSR/
├── LQ_proj_in.ckpt
├── TCDecoder.ckpt
├── diffusion_pytorch_model_streaming_dmd.safetensors
└── Wan2.1_VAE.pth

基本使用

参数说明

参数	说明
mode	tiny 速度快（默认），full 质量高
scale	建议使用 4，VRAM 不足时用 2
color_fix	使用小波变换修正视频色彩
tiled_vae	设为 True 可降低解码时的 VRAM 占用
tiled_dit	显著降低 VRAM 占用，但速度变慢
tile_size	分块大小
tile_overlap	分块重叠区域
unload_dit	解码前卸载 DiT，降低 VRAM 峰值

最简使用示例

在 ComfyUI 中加载 FlashVSR Ultra Fast 节点，输入视频并连接输出即可。使用默认参数即可获得不错的效果。

如遇 VRAM 不足，依次尝试：

1. 将 scale 改为 2
2. 开启 tiled_dit
3. 开启 tiled_vae
4. 开启 unload_dit