xinyun/ComfyUI-KJNodes
1
0
Fork 0
mirror of https://git.datalinker.icu/kijai/ComfyUI-KJNodes.git synced 2026-05-25 20:49:07 +08:00
Code Issues Packages Projects Releases Wiki Activity

Update model_optimization_nodes.py

Browse Source
This commit is contained in:
kijai 2025-12-03 17:18:13 +02:00
parent 37206374ef
commit 50e7dd34d3
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
nodes/model_optimization_nodes.py
View File

@ -2237,7 +2237,7 @@ class EndRecordCUDAMemoryHistory():
RETURN_NAMES = ("input", "output_path",) RETURN_NAMES = ("input", "output_path",)
FUNCTION = "end" FUNCTION = "end"
CATEGORY = "KJNodes/experimental" CATEGORY = "KJNodes/experimental"
DESCRIPTION = "Records CUDA memory allocation history between start and end, saves to a file that can be analyzed here: https://docs.pytorch.org/memory_viz" DESCRIPTION = "Records CUDA memory allocation history between start and end, saves to a file that can be analyzed here: https://docs.pytorch.org/memory_viz or with VisualizeCUDAMemoryHistory node"
def end(self, input, output_path): def end(self, input, output_path):
mm.soft_empty_cache() mm.soft_empty_cache()