From 1040f0950d29751d0b3761d9766b7fb8a74ad56b Mon Sep 17 00:00:00 2001
From: kijai <40791699+kijai@users.noreply.github.com>
Date: Mon, 26 Feb 2024 21:56:14 +0200
Subject: [PATCH] Update nodes.py

---
 nodes.py | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/nodes.py b/nodes.py
index a0f5222..b04bece 100644
--- a/nodes.py
+++ b/nodes.py
@@ -1157,13 +1157,14 @@ class VRAM_Debug:
 
     def VRAMdebug(self, image_passthrough, empty_cache, unload_all_models):
         freemem_before = comfy.model_management.get_free_memory()
-        print(freemem_before)
+        print("VRAMdebug: free memory before: ", freemem_before)
         if empty_cache:
             comfy.model_management.soft_empty_cache()
         if unload_all_models:
             comfy.model_management.unload_all_models()
         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)
 
 class AnyType(str):