wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-01-24 05:10:18 +08:00
Code Issues Actions 13 Packages Projects Releases Wiki Activity

Merge branch 'comfyanonymous:master' into master

Browse Source
This commit is contained in:
patientx 2025-11-12 17:55:30 +03:00 committed by GitHub
commit 644778be49
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 6 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
comfy/model_management.py
View File

@ -1108,6 +1108,9 @@ def pin_memory(tensor):
if MAX_PINNED_MEMORY <= 0: if MAX_PINNED_MEMORY <= 0:
return False return False
if type(tensor) is not torch.nn.parameter.Parameter:
return False
if not is_device_cpu(tensor.device): if not is_device_cpu(tensor.device):
return False return False
@ -1117,6 +1120,9 @@ def pin_memory(tensor):
#on the GPU async. So dont trust the CUDA API and guard here #on the GPU async. So dont trust the CUDA API and guard here
return False return False
if not tensor.is_contiguous():
return False
size = tensor.numel() * tensor.element_size() size = tensor.numel() * tensor.element_size()
if (TOTAL_PINNED_MEMORY + size) > MAX_PINNED_MEMORY: if (TOTAL_PINNED_MEMORY + size) > MAX_PINNED_MEMORY:
return False return False