nodes/Isi-dev-ComfyUI_DeleteModelPassthrough
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Update delete_model_passthrough.py

Browse Source
This commit is contained in:
Isi
2025-08-29 12:37:34 +01:00
committed by GitHub
parent 582e0687b5
commit d931cb5150
1 changed files with 10 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files
delete_model_passthrough.py
+10 -10
View File
@@ -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")