xinyun/ComfyUI-KJNodes
1
0
Fork 0
mirror of https://git.datalinker.icu/kijai/ComfyUI-KJNodes.git synced 2025-12-15 07:44:30 +08:00
Code Issues Packages Projects Releases Wiki Activity

VRAM debug node partial rollback

Browse Source
This commit is contained in:
kijai 2024-04-11 01:50:45 +03:00
parent 02baeb5248
commit 6f53738fe6
1 changed files with 12 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
nodes.py
View File

@ -1290,24 +1290,29 @@ class VRAM_Debug:
def INPUT_TYPES(s): def INPUT_TYPES(s):
return { return {
"required": { "required": {
"input": (any, {}),
"empty_cache": ("BOOLEAN", {"default": True}), "empty_cache": ("BOOLEAN", {"default": True}),
"gc_collect": ("BOOLEAN", {"default": True}), "gc_collect": ("BOOLEAN", {"default": True}),
"unload_all_models": ("BOOLEAN", {"default": False}), "unload_all_models": ("BOOLEAN", {"default": False}),
}, },
"optional": {
"any_input": (any, {}),
"image_pass": ("IMAGE",),
"model_pass": ("MODEL",),
}
} }
RETURN_TYPES = (any, "INT", "INT",) RETURN_TYPES = (any, "IMAGE","MODEL","INT", "INT",)
RETURN_NAMES = ("output", "freemem_before", "freemem_after") RETURN_NAMES = ("any_output", "image_pass", "model_pass", "freemem_before", "freemem_after")
FUNCTION = "VRAMdebug" FUNCTION = "VRAMdebug"
CATEGORY = "KJNodes/misc" CATEGORY = "KJNodes/misc"
DESCRIPTION = """ DESCRIPTION = """
Returns the input unchanged, and performs comfy model Returns the inputs unchanged, they are only used as triggers,
management functions and garbage collection, and performs comfy model management functions and garbage collection,
reports free VRAM before and after the operations. reports free VRAM before and after the operations.
""" """
def VRAMdebug(self, input, gc_collect,empty_cache, unload_all_models): def VRAMdebug(self, gc_collect,empty_cache, unload_all_models, image_pass=None, model_pass=None, any_input=None):
freemem_before = model_management.get_free_memory() freemem_before = model_management.get_free_memory()
print("VRAMdebug: free memory before: ", freemem_before) print("VRAMdebug: free memory before: ", freemem_before)
if empty_cache: if empty_cache:
@ -1320,7 +1325,7 @@ reports free VRAM before and after the operations.
freemem_after = model_management.get_free_memory() freemem_after = model_management.get_free_memory()
print("VRAMdebug: free memory after: ", freemem_after) print("VRAMdebug: free memory after: ", freemem_after)
print("VRAMdebug: freed memory: ", freemem_after - freemem_before) print("VRAMdebug: freed memory: ", freemem_after - freemem_before)
return (input, freemem_before, freemem_after) return (any_input, image_pass, model_pass, freemem_before, freemem_after)
class SomethingToString: class SomethingToString:
@classmethod @classmethod