Skip to content

Commit 0f4ef3a

Browse files
This seems to slow things down slightly on Linux. (Comfy-Org#10624)
1 parent 6b88478 commit 0f4ef3a

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

comfy/ops.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -35,7 +35,7 @@ def scaled_dot_product_attention(q, k, v, *args, **kwargs):
3535

3636

3737
try:
38-
if torch.cuda.is_available():
38+
if torch.cuda.is_available() and comfy.model_management.WINDOWS:
3939
from torch.nn.attention import SDPBackend, sdpa_kernel
4040
import inspect
4141
if "set_priority" in inspect.signature(sdpa_kernel).parameters:

0 commit comments

Comments
 (0)