update

ldm/modules/attention.py

@@ -211,7 +211,7 @@ class CrossAttention(nn.Module):
         dim_head=64,
         dropout=0.0,
         # backend=None,
-        backend=SDPBackend.FLASH_ATTENTION, # FA implemented by torch.
+        # backend=SDPBackend.FLASH_ATTENTION, # FA implemented by torch.
         **kwargs,
     ):
         super().__init__()
@@ -228,7 +228,7 @@ class CrossAttention(nn.Module):
         self.to_out = nn.Sequential(
             nn.Linear(inner_dim, query_dim), nn.Dropout(dropout)
         )
-        self.backend = backend
+        # self.backend = backend
 
     def forward(
         self,