Unverified Commit 82c795d6 authored by Huamin Li's avatar Huamin Li Committed by GitHub
Browse files

Fix AttributeError about _use_fi_prefill (#29734)


Signed-off-by: default avatarHuamin Li <3ericli@gmail.com>
parent e1464c3a
...@@ -2061,7 +2061,7 @@ class MLACommonImpl(MLACommonBaseImpl[M], Generic[M]): ...@@ -2061,7 +2061,7 @@ class MLACommonImpl(MLACommonBaseImpl[M], Generic[M]):
attn_out, attn_out,
lse, lse,
get_dcp_group(), get_dcp_group(),
is_lse_base_on_e=not self._use_fi_prefill, is_lse_base_on_e=not getattr(self, "_use_fi_prefill", False),
) )
# v_up projection # v_up projection
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment