From 9a33651bb5d51a762eeccffce86d53d4e6eca68f Mon Sep 17 00:00:00 2001
From: Sakura-Luna <53183413+Sakura-Luna@users.noreply.github.com>
Date: Sun, 29 Jun 2025 16:21:57 +0800
Subject: [PATCH] Update attention.py

Fix https://github.com/comfyanonymous/ComfyUI/issues/8634
---
 comfy/ldm/modules/attention.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/comfy/ldm/modules/attention.py b/comfy/ldm/modules/attention.py
index 35d2270ee..54bfc84e8 100644
--- a/comfy/ldm/modules/attention.py
+++ b/comfy/ldm/modules/attention.py
@@ -446,6 +446,7 @@ def attention_pytorch(q, k, v, heads, mask=None, attn_precision=None, skip_resha
         # add a heads dimension if there isn't already one
         if mask.ndim == 3:
             mask = mask.unsqueeze(1)
+        mask = mask.contiguous()
 
     if SDP_BATCH_LIMIT >= b:
         out = torch.nn.functional.scaled_dot_product_attention(q, k, v, attn_mask=mask, dropout_p=0.0, is_causal=False)