jaidaken/ComfyUI
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

Add some warnings for pin and unpin errors. (#11561)

Browse Source
This commit is contained in:
comfyanonymous
2025-12-29 15:26:42 -08:00
committed by GitHub
parent 9ca7e143af
commit 0e6221cc79
1 changed files with 2 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
comfy/model_management.py
View File

@@ -1169,6 +1169,7 @@ def pin_memory(tensor):
TOTAL_PINNED_MEMORY += size TOTAL_PINNED_MEMORY += size
return True return True
else: else:
logging.warning("Pin error.")
discard_cuda_async_error() discard_cuda_async_error()
return False return False
@@ -1199,6 +1200,7 @@ def unpin_memory(tensor):
TOTAL_PINNED_MEMORY = 0 TOTAL_PINNED_MEMORY = 0
return True return True
else: else:
logging.warning("Unpin error.")
discard_cuda_async_error() discard_cuda_async_error()
return False return False