From fe6749102cef8fbb1dc7ea33da17c97a5a1376b5 Mon Sep 17 00:00:00 2001 From: Isi Date: Thu, 28 Aug 2025 18:39:08 +0100 Subject: [PATCH] Update delete_model_passthrough.py --- delete_model_passthrough.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/delete_model_passthrough.py b/delete_model_passthrough.py index e81688e..38562e1 100644 --- a/delete_model_passthrough.py +++ b/delete_model_passthrough.py @@ -44,6 +44,7 @@ class DeleteModelPassthrough: if not torch.cuda.is_available(): return "CUDA not available." initial = torch.cuda.memory_allocated() + print(f"Initial GPU memory: {initial}") try: del model @@ -54,6 +55,7 @@ class DeleteModelPassthrough: print(f"Failed to delete model: {e}") final = torch.cuda.memory_allocated() + print(f"Final GPU memory: {final}") freed = max(0, initial - final) print(f"GPU VRAM freed: {freed/1e9:.2f} GB")