This seems to slow things down slightly on Linux.

This commit is contained in:
comfyanonymous 2025-11-03 21:39:15 -05:00 committed by GitHub
parent 6b88478f9f
commit 499f6d8a82
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -35,7 +35,7 @@ def scaled_dot_product_attention(q, k, v, *args, **kwargs):
try:
if torch.cuda.is_available():
if torch.cuda.is_available() and comfy.model_management.WINDOWS:
from torch.nn.attention import SDPBackend, sdpa_kernel
import inspect
if "set_priority" in inspect.signature(sdpa_kernel).parameters: