pfnet
/

plamo-embedding-1b

Sentence Similarity

feature-extraction

Model card Files Files and versions

kaitos255 commited on 5 days ago

Commit

2fcf411

·

verified ·

1 Parent(s): d25181b

Fix behavior when batch_size=1

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,
 }