From 119941174704081a16a4c3f303d99f2fb1e95cde Mon Sep 17 00:00:00 2001
From: comfyanonymous <121283862+comfyanonymous@users.noreply.github.com>
Date: Tue, 11 Nov 2025 16:33:30 -0800
Subject: [PATCH] Don't pin tensor if not a torch.nn.parameter.Parameter
 (#10718)

---
 comfy/model_management.py | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/comfy/model_management.py b/comfy/model_management.py
index a4410f2ec..d8913082a 100644
--- a/comfy/model_management.py
+++ b/comfy/model_management.py
@@ -1107,6 +1107,9 @@ def pin_memory(tensor):
     if MAX_PINNED_MEMORY <= 0:
         return False
 
+    if type(tensor) is not torch.nn.parameter.Parameter:
+        return False
+
     if not is_device_cpu(tensor.device):
         return False
 
@@ -1116,6 +1119,9 @@ def pin_memory(tensor):
         #on the GPU async. So dont trust the CUDA API and guard here
         return False
 
+    if not tensor.is_contiguous():
+        return False
+
     size = tensor.numel() * tensor.element_size()
     if (TOTAL_PINNED_MEMORY + size) > MAX_PINNED_MEMORY:
         return False