Disable non_blocking when --deterministic or directml.

main
comfyanonymous 9 months ago
parent 71ec5b144e
commit bf3e334d46

@ -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):

Loading…
Cancel
Save