Use stream for casting if enabled. (comfyanonymous#7833)

comfyanonymous · web-flow · commit c8cd7ad795ec · 2025-04-27T05:38:11.000-04:00
diff --git a/comfy/model_management.py b/comfy/model_management.py
@@ -980,6 +980,9 @@ def cast_to(weight, dtype=None, device=None, non_blocking=False, copy=False, str
         if not copy:
             if dtype is None or weight.dtype == dtype:
                 return weight
+        if stream is not None:
+            with stream:
+                return weight.to(dtype=dtype, copy=copy)
         return weight.to(dtype=dtype, copy=copy)
 
     if stream is not None:
