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]):
attn_out,
lse,
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
......
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