Fix merge

vllm-project · Nov 1, 2024 · 821d288 · 821d288
1 parent 16ee580
commit 821d288
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/vllm/attention/backends/flashinfer.py b/vllm/attention/backends/flashinfer.py
@@ -871,7 +871,7 @@ def unified_flash_infer(
         else:
             assert prefill_meta is not None
             assert prefill_meta.prefill_wrapper is not None
-            output = prefill_meta.prefill_wrapper.forward(
+            prefill_output = prefill_meta.prefill_wrapper.forward(
                 query,
                 kv_cache,
                 logits_soft_cap=logits_soft_cap,