wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-02-01 01:00:20 +08:00
Code Issues Actions 22 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 17cdb0284b
commit 645c4597d2
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
size = module._pin.numel() * module._pin.element_size()
comfy.model_management.unpin_memory(module._pin)
del module._pin
return size