mirror of
https://github.com/comfyanonymous/ComfyUI.git
synced 2026-02-08 04:22:36 +08:00
Merge 31b27afe69 into b8315e66cb
This commit is contained in:
commit
1c1a6d83d4
@ -386,7 +386,7 @@ except:
|
||||
pass
|
||||
|
||||
|
||||
if ENABLE_PYTORCH_ATTENTION:
|
||||
if ENABLE_PYTORCH_ATTENTION and cpu_state != CPUState.CPU:
|
||||
torch.backends.cuda.enable_math_sdp(True)
|
||||
torch.backends.cuda.enable_flash_sdp(True)
|
||||
torch.backends.cuda.enable_mem_efficient_sdp(True)
|
||||
|
||||
Loading…
Reference in New Issue
Block a user