Fix behavior when batch_size=1

modeling_plamo.py

@@ -817,7 +817,7 @@ class ModifiedAttention(Attention):
 
 
 PLAMO_ATTENTION_CLASSES = {
-    "sdpa": ModifiedAttention,
+    "sdpa": Attention,
 }