Disable non_blocking when --deterministic or directml.

comfyanonymous · comfyanonymous · commit bf3e334d4688 · 2024-05-30T11:07:38.000-04:00
diff --git a/comfy/model_management.py b/comfy/model_management.py
@@ -629,6 +629,10 @@ def supports_dtype(device, dtype): #TODO
 def device_supports_non_blocking(device):
     if is_device_mps(device):
         return False #pytorch bug? mps doesn't support non blocking
+    if args.deterministic: #TODO: figure out why deterministic breaks non blocking from gpu to cpu (previews)
+        return False
+    if directml_enabled:
+        return False
     return True
 
 def device_should_use_non_blocking(device):
