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

Update nodes.py

Browse Source
This commit is contained in:
kijai 2024-02-26 21:56:14 +02:00
parent 1b9d469830
commit 1040f0950d
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
nodes.py
View File

@ -1157,13 +1157,14 @@ class VRAM_Debug:
def VRAMdebug(self, image_passthrough, empty_cache, unload_all_models): def VRAMdebug(self, image_passthrough, empty_cache, unload_all_models):
freemem_before = comfy.model_management.get_free_memory() freemem_before = comfy.model_management.get_free_memory()
print(freemem_before) print("VRAMdebug: free memory before: ", freemem_before)
if empty_cache: if empty_cache:
comfy.model_management.soft_empty_cache() comfy.model_management.soft_empty_cache()
if unload_all_models: if unload_all_models:
comfy.model_management.unload_all_models() comfy.model_management.unload_all_models()
freemem_after = comfy.model_management.get_free_memory() freemem_after = comfy.model_management.get_free_memory()
print(freemem_after) print("VRAMdebug: free memory after: ", freemem_after)
print("VRAMdebug: freed memory: ", freemem_after - freemem_before)
return (image_passthrough, freemem_before, freemem_after) return (image_passthrough, freemem_before, freemem_after)
class AnyType(str): class AnyType(str):