Skip to content

Commit 569cc78

Browse files
committed
fix param name
Signed-off-by: jiqing-feng <jiqing.feng@intel.com>
1 parent 422134f commit 569cc78

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

optimum/exporters/ipex/modeling_utils.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -617,8 +617,8 @@ def forward(
617617
PagedAttention.flash_attn_varlen_func(
618618
attn_output,
619619
query,
620-
key_cache,
621-
value_cache,
620+
key,
621+
value,
622622
seq_len_tensor,
623623
seq_len_tensor,
624624
input_lens.max(),

0 commit comments

Comments
 (0)