mirror of
https://github.com/comfyanonymous/ComfyUI.git
synced 2026-01-16 09:10:50 +08:00
add missing del on unpin
This commit is contained in:
parent
e2d29926a9
commit
2c554a112e
@ -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
|
||||
|
||||
Loading…
Reference in New Issue
Block a user