From b99417fd4a4788ea57de479ac3fec14531415d8d Mon Sep 17 00:00:00 2001 From: kijai <40791699+kijai@users.noreply.github.com> Date: Thu, 1 Aug 2024 11:17:07 +0300 Subject: [PATCH] Update transformer.py --- sam2/modeling/sam/transformer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/sam2/modeling/sam/transformer.py b/sam2/modeling/sam/transformer.py index 2dc5fbc..fb71606 100644 --- a/sam2/modeling/sam/transformer.py +++ b/sam2/modeling/sam/transformer.py @@ -331,7 +331,7 @@ class RoPEAttention(Attention): # Attention if not OLD_TORCH: if not MATH_KERNEL_ON and OLD_GPU and dropout_p > 0.0: - backends.append(SDPBackend.MATH)) + backends.append(SDPBackend.MATH) with sdpa_kernel(backends): out = F.scaled_dot_product_attention(q, k, v, dropout_p=dropout_p) else: