Skip to content

Commit a7ca264

Browse files
committed
fix missing args
Signed-off-by: Wuxun Zhang <[email protected]>
1 parent 4f32ae1 commit a7ca264

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

vllm_gaudi/v1/worker/hpu_model_runner.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2949,6 +2949,8 @@ def execute_model(
29492949
attn_metadata,
29502950
logits_indices,
29512951
self.kv_caches,
2952+
None,
2953+
None,
29522954
warmup_mode=warmup_mode)
29532955
htorch.core.mark_step()
29542956

@@ -3068,6 +3070,8 @@ def execute_model(
30683070
dummy_decode_input_data_across_dp.attn_metadata,
30693071
dummy_decode_input_data_across_dp.logits_indices,
30703072
self.kv_caches,
3073+
None,
3074+
None,
30713075
warmup_mode=warmup_mode)
30723076
htorch.core.mark_step()
30733077

0 commit comments

Comments
 (0)