功能建议
在 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("模型卸载失败")
优点
- 用户能清楚了解内存清理的实际效果
- 方便排查问题(例如模型未卸载的原因)
- 不影响现有功能,只是增加日志输出
功能建议
在 VRAMCleanup 节点执行模型卸载时,增加模型卸载状态的输出信息,方便用户确认是否成功卸载。
当前行为
调用
unload_all_models()后,只能看到简单的日志输出:无法确认模型是否真正被卸载。
期望行为
增加明确的卸载状态提示,例如:
模型卸载成功模型卸载失败或更详细的信息:
[✓] 模型卸载成功(0个模型加载)[✗] 模型卸载失败(X个模型仍在加载)实现方案
调用
unload_all_models()后,使用comfy.model_management.loaded_models()检查当前加载的模型:优点