Windecay/ComfyUI-ReservedVRAM

A simple node that can dynamically adjust the reserved memory of a workflow in real-time, used to avoid the utilization of shared memory.

解决的问题

该项目提供了一个ComfyUI节点,允许用户实时动态调整工作流预留的VRAM量。这有助于防止内存不足(OOM)错误,并管理Windows任务管理器中的共享内存溢出,让用户无需担心内存限制即可最大化GPU性能。

工作原理

该节点与ComfyUI的内存管理系统集成。它可以更新EXTRA_RESERVED_VRAM环境变量,或在可用时使用model_management.set_extra_reserved_vram()运行时API。它还支持与DynamicVRAM和comfy-aimdo的兼容性,以同步预留空间设置。在“自动”模式下,节点检测系统中“已使用”的VRAM量,并添加用户定义的值来计算预留量。

适用人群

需要精确控制GPU内存以避免崩溃或优化性能的ComfyUI用户,尤其是运行多个进程或使用高分辨率生成的用户。

亮点

  • 实时调整: 更改预留的VRAM值(以GB为单位),并在工作流执行期间立即生效。
  • 自动模式: 自动检测系统VRAM使用情况,防止进程挂起。
  • VRAM清理: 包含一个开关,在输出前清除VRAM,有效充当内存清理节点。
  • 随机种子集成: 可以充当随机种子节点,每次运行都修改VRAM策略。
  • 广泛兼容性: 适用于官方ComfyUI构建以及DynamicVRAM和comfy-aimdo等特定扩展。

相关

  • 项目
  • 项目
  • 项目
  • 项目