Skip to content

[建议] VRAM清理时增加模型卸载状态的输出信息 #8

@LWWZH

Description

@LWWZH

功能建议

在 VRAMCleanup 节点执行模型卸载时,增加模型卸载状态的输出信息,方便用户确认是否成功卸载。

当前行为

调用 unload_all_models() 后,只能看到简单的日志输出:

VRAM清理完成 [卸载模型: True, 清空缓存: True]

无法确认模型是否真正被卸载。

期望行为

增加明确的卸载状态提示,例如:

  • 卸载成功:模型卸载成功
  • 卸载失败:模型卸载失败

或更详细的信息:

  • 卸载成功:[✓] 模型卸载成功(0个模型加载)
  • 卸载失败:[✗] 模型卸载失败(X个模型仍在加载)

实现方案

调用 unload_all_models() 后,使用 comfy.model_management.loaded_models() 检查当前加载的模型:

if offload_model:
    comfy.model_management.unload_all_models()
    if len(comfy.model_management.loaded_models()) == 0:
        print("模型卸载成功")
    else:
        print("模型卸载失败")

优点

  • 用户能清楚了解内存清理的实际效果
  • 方便排查问题(例如模型未卸载的原因)
  • 不影响现有功能,只是增加日志输出

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions