xinyun/ComfyUI
1
0
Fork 0
mirror of https://git.datalinker.icu/comfyanonymous/ComfyUI synced 2026-08-16 05:36:37 +08:00
Code Issues Packages Projects Releases Wiki Activity

Disable non_blocking when --deterministic or directml.

Browse Source
This commit is contained in:
comfyanonymous 2024-05-30 11:07:38 -04:00
parent 228c1e3354
commit 9cea5b91a0
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
comfy/model_management.py
View File

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