add missing del on unpin

2026-03-15 22:27:34 +08:00 · 2026-01-13 19:39:00 +10:00 · 2026-01-13 19:39:00 +10:00 · e8c9977973
commit e8c9977973
parent e2d62b8f80
1 changed files with 1 additions and 0 deletions
--- a/comfy/pinned_memory.py
+++ b/comfy/pinned_memory.py
@ -31,4 +31,5 @@ def unpin_memory(module):
        return 0
    size = module._pin.numel() * module._pin.element_size()
    comfy.model_management.unpin_memory(module._pin)
+    del module._pin
    return size