Skip to content

GitLab

"git@developer.sourcefind.cn:chenpangpang/ComfyUI.git" did not exist on "0aa667ed33aae800880153a91c283ac457d0b31c"
Commit bf3e334d authored by comfyanonymous's avatar comfyanonymous
Browse files

Disable non_blocking when --deterministic or directml.

parent 71ec5b14
Show whitespace changes
Inline Side-by-side
Showing with 4 additions and 0 deletions
comfy/model_management.py
View file @ bf3e334d
...@@ -629,6 +629,10 @@ def supports_dtype(device, dtype): #TODO ...@@ -629,6 +629,10 @@ def supports_dtype(device, dtype): #TODO
def device_supports_non_blocking(device): def device_supports_non_blocking(device):
if is_device_mps(device): if is_device_mps(device):
return False #pytorch bug? mps doesn't support non blocking 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 return True
def device_should_use_non_blocking(device): def device_should_use_non_blocking(device):
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment