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等特定扩展。
相关
- 项目
- 项目
- 项目
- 项目