This seems to slow things down slightly on Linux. (#10624)

This commit is contained in:
comfyanonymous 2025-11-03 18:47:14 -08:00 committed by GitHub
parent 6b88478f9f
commit 0f4ef3afa0
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: