From 64937567ffce7ba9dfe09c4ed0adc5e4ddcd0c2b Mon Sep 17 00:00:00 2001
From: comfyanonymous <121283862+comfyanonymous@users.noreply.github.com>
Date: Sun, 18 Aug 2024 14:34:03 -0400
Subject: [PATCH] Update ops.py

---
 comfy/ops.py | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/comfy/ops.py b/comfy/ops.py
index 139307cfe..d76138c92 100644
--- a/comfy/ops.py
+++ b/comfy/ops.py
@@ -21,8 +21,7 @@ import comfy.model_management
 
 
 def cast_to(weight, dtype=None, device=None, non_blocking=False):
-    if dtype == torch.float32:
-        non_blocking = False
+    non_blocking = False
     return weight.to(device=device, dtype=dtype, non_blocking=non_blocking)
 
 def cast_to_input(weight, input, non_blocking=False):