diff --git a/comfy/pinned_memory.py b/comfy/pinned_memory.py index be303b4f1..923872dac 100644 --- 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