add missing del on unpin

This commit is contained in:
Rattus 2026-01-13 19:39:00 +10:00
parent e2d29926a9
commit 2c554a112e

View File

@ -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