kaitos255 commited on
Commit
5f5fbe6
·
verified ·
1 Parent(s): d25181b

Fix behavior when batch_size=1 (#1)

Browse files

- Fix behavior when batch_size=1 (2fcf4114d0ff2b5a88823f1e6c1deff57493d911)

Files changed (1) hide show
  1. modeling_plamo.py +1 -1
modeling_plamo.py CHANGED
@@ -817,7 +817,7 @@ class ModifiedAttention(Attention):
817
 
818
 
819
  PLAMO_ATTENTION_CLASSES = {
820
- "sdpa": ModifiedAttention,
821
  }
822
 
823
 
 
817
 
818
 
819
  PLAMO_ATTENTION_CLASSES = {
820
+ "sdpa": Attention,
821
  }
822
 
823