mirror of
https://github.com/comfyanonymous/ComfyUI.git
synced 2026-02-02 09:40:19 +08:00
add missing del on unpin
This commit is contained in:
parent
6e852baa9a
commit
15ae09fb19
@ -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