wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-02-07 12:02:37 +08:00
Code Issues Actions 15 Packages Projects Releases Wiki Activity

add missing del on unpin

Browse Source
This commit is contained in:
Rattus 2026-01-13 19:39:00 +10:00
parent 6e852baa9a
commit 15ae09fb19
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
comfy/pinned_memory.py
View File

@ -31,4 +31,5 @@ def unpin_memory(module):
return 0 return 0
size = module._pin.numel() * module._pin.element_size() size = module._pin.numel() * module._pin.element_size()
comfy.model_management.unpin_memory(module._pin) comfy.model_management.unpin_memory(module._pin)
del module._pin
return size return size