From d931cb5150535b36f719fc1eba735cad338e0a2d Mon Sep 17 00:00:00 2001 From: Isi Date: Fri, 29 Aug 2025 12:37:34 +0100 Subject: [PATCH] Update delete_model_passthrough.py --- delete_model_passthrough.py | 20 ++++++++++---------- 1 file changed, 10 insertions(+), 10 deletions(-) diff --git a/delete_model_passthrough.py b/delete_model_passthrough.py index b20df5b..3a7843e 100644 --- a/delete_model_passthrough.py +++ b/delete_model_passthrough.py @@ -372,24 +372,24 @@ class DeleteModelPassthrough: after_ram = psutil.virtual_memory().percent # Log results - print("=" * 60) - print("📊 Memory Free Results:") - print("=" * 60) - print(f"🗑️ Deleted model type: {deleted_type}") - print(f"💾 System RAM change: {before_ram - after_ram:+.2f}%") + # print("=" * 60) + # print("📊 Memory Free Results:") + # print("=" * 60) + # print(f"🗑️ Deleted model type: {deleted_type}") + # print(f"💾 System RAM change: {before_ram - after_ram:+.2f}%") if torch.cuda.is_available(): vram_freed = (before_vram - after_vram) / (1024 * 1024 * 1024) reserved_freed = (before_reserved - after_reserved) / (1024 * 1024 * 1024) - print(f"🎮 GPU allocated freed: {vram_freed:.3f} GB") - print(f"🎮 GPU reserved freed: {reserved_freed:.3f} GB") - print(f"🎮 Final allocated: {after_vram / (1024 * 1024 * 1024):.3f} GB") - print(f"🎮 Final reserved: {after_reserved / (1024 * 1024 * 1024):.3f} GB") + # print(f"🎮 GPU allocated freed: {vram_freed:.3f} GB") + # print(f"🎮 GPU reserved freed: {reserved_freed:.3f} GB") + # print(f"🎮 Final allocated: {after_vram / (1024 * 1024 * 1024):.3f} GB") + # print(f"🎮 Final reserved: {after_reserved / (1024 * 1024 * 1024):.3f} GB") print("=" * 60) # Success determination based on memory freed - if torch.cuda.is_available() and reserved_freed > 0.1: # Freed at least 100MB + if torch.cuda.is_available() and reserved_freed > 1: # Freed at least 1000MB print("✅ SUCCESS: Significant memory freed!") elif torch.cuda.is_available() and reserved_freed > 0: print("⚠️ PARTIAL: Some memory freed, but model may still be referenced")