diff --git a/comfy/model_management.py b/comfy/model_management.py index 5c1afd3..b353e50 100644 --- 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):