Skip to content

Commit 8aa5978

Browse files
committed
change condition
1 parent 72cd3c8 commit 8aa5978

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

optimum/exporters/openvino/model_patcher.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1613,7 +1613,7 @@ def select_ext_factor(
16131613
seq_len: torch.Tensor, max_pos_embeddings: torch.Tensor, short_factor: torch.Tensor, long_factor: torch.Tensor
16141614
):
16151615
return torch.where(
1616-
seq_len < max_pos_embeddings, short_factor, long_factor
1616+
seq_len <= max_pos_embeddings, short_factor, long_factor
16171617
) # short_factor * (seq_len <= max_pos_embeddings) + long_factor * (seq_len > max_pos_embeddings)
16181618

16191619

0 commit comments

Comments
 (0)