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

Merge 8d8e0c68b0 into 985fb9d6ad

Browse Source
This commit is contained in:
liminfei-amd 2026-07-06 09:06:47 +08:00 committed by GitHub
commit 84f8a070a3
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 8 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
comfy/model_management.py
View File

@ -1534,7 +1534,14 @@ def pin_memory(tensor):
size = tensor.nbytes size = tensor.nbytes
comfy.memory_management.extra_ram_release(comfy.memory_management.RAM_CACHE_HEADROOM) comfy.memory_management.extra_ram_release(comfy.memory_management.RAM_CACHE_HEADROOM)
ensure_pin_registerable(size) # Respect the host-RAM budget like the dynamic-VRAM pin path (comfy/pinned_memory.py)
# already does. Without this gate the static load path keeps cudaHostRegister-ing
# offloaded weights toward MAX_PINNED_MEMORY (90% of RAM on Linux) regardless of how
# little RAM is actually free, so unswappable pages fill RAM+swap and large model
# loads stall (issue #13730). When the budget cannot be met, skip pinning and leave
# the weight in pageable RAM (still correct, just not pinned).
if not ensure_pin_budget(size) or not ensure_pin_registerable(size):
return False
ptr = tensor.data_ptr() ptr = tensor.data_ptr()
if ptr == 0: if ptr == 0: