Fix behavior when batch_size=1 (#1)
Browse files- Fix behavior when batch_size=1 (2fcf4114d0ff2b5a88823f1e6c1deff57493d911)
- modeling_plamo.py +1 -1
modeling_plamo.py
CHANGED
@@ -817,7 +817,7 @@ class ModifiedAttention(Attention):
|
|
817 |
|
818 |
|
819 |
PLAMO_ATTENTION_CLASSES = {
|
820 |
-
"sdpa":
|
821 |
}
|
822 |
|
823 |
|
|
|
817 |
|
818 |
|
819 |
PLAMO_ATTENTION_CLASSES = {
|
820 |
+
"sdpa": Attention,
|
821 |
}
|
822 |
|
823 |
|