Fix AttributeError about _use_fi_prefill (#29734)

hl475 · web-flow · commit 82c795d6f28e · 2025-11-30T06:04:55.000Z
Signed-off-by: Huamin Li &lt;3ericli@gmail.com&gt;
diff --git a/vllm/v1/attention/backends/mla/common.py b/vllm/v1/attention/backends/mla/common.py
@@ -2061,7 +2061,7 @@ def forward(
                     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

Original file line number	Diff line number	Diff line change
`@@ -2061,7 +2061,7 @@ def forward(`
`2061`	`2061`	`attn_out,`
`2062`	`2062`	`lse,`
`2063`	`2063`	`get_dcp_group(),`
`2064`		`- is_lse_base_on_e=not self._use_fi_prefill,`
	`2064`	`+ is_lse_base_on_e=not getattr(self, "_use_fi_prefill", False),`
`2065`	`2065`	`)`
`2066`	`2066`
`2067`	`2067`	`# v_up projection`