From c4e65114646871b23633246831730883aa32be0e Mon Sep 17 00:00:00 2001
From: Rattus <rattus128@gmail.com>
Date: Mon, 30 Mar 2026 00:34:21 +1000
Subject: [PATCH] mm: Remove pin count clearing threshold.

TOTAL_PINNED_MEMORY is shared between the legacy and aimdo pinning
systems, however this catch-all assumes only the legacy system exists.
Remove the catch-all as the PINNED_MEMORY buffer is coherent already.
---
 comfy/model_management.py | 2 --
 1 file changed, 2 deletions(-)

diff --git a/comfy/model_management.py b/comfy/model_management.py
index c1b094486..0eebf1ded 100644
--- a/comfy/model_management.py
+++ b/comfy/model_management.py
@@ -1403,8 +1403,6 @@ def unpin_memory(tensor):
 
     if torch.cuda.cudart().cudaHostUnregister(ptr) == 0:
         TOTAL_PINNED_MEMORY -= PINNED_MEMORY.pop(ptr)
-        if len(PINNED_MEMORY) == 0:
-            TOTAL_PINNED_MEMORY = 0
         return True
     else:
         logging.warning("Unpin error.")