diff --git a/comfy/conds.py b/comfy/conds.py
index 775785d37..e643d6830 100644
--- a/comfy/conds.py
+++ b/comfy/conds.py
@@ -18,7 +18,10 @@ class CONDRegular:
             return cond
 
     def process_cond(self, batch_size, device, **kwargs):
-        return self._copy_with(comfy.utils.repeat_to_batch_size(self._pin_memory(self.cond), batch_size).to(device, non_blocking=device_should_use_non_blocking(device))
+        if device_should_use_non_blocking(device):
+            return self._copy_with(comfy.utils.repeat_to_batch_size(self._pin_memory(self.cond), batch_size).to(device, non_blocking=True)
+        else:
+            return self._copy_with(comfy.utils.repeat_to_batch_size(self.cond, batch_size).to(device)
 
     def can_concat(self, other):
         if self.cond.shape != other.cond.shape: