pfnet
/

plamo-embedding-1b

Sentence Similarity

feature-extraction

Model card Files Files and versions

kaitos255 commited on Aug 8

Commit

5f5fbe6

·

verified ·

1 Parent(s): d25181b

Fix behavior when batch_size=1 (#1)

- Fix behavior when batch_size=1 (2fcf4114d0ff2b5a88823f1e6c1deff57493d911)

Files changed (1) hide show

modeling_plamo.py +1 -1

modeling_plamo.py CHANGED Viewed

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

 PLAMO_ATTENTION_CLASSES = {
+    "sdpa": Attention,
 }