Skip to content

Commit 127f0ba

Browse files
authored
Apply suggestions from code review
1 parent 72cd3c8 commit 127f0ba

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

optimum/exporters/openvino/model_patcher.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1608,13 +1608,12 @@ def _phi3_self_attn_sdpa_forward(
16081608
return attn_output, None, past_key_value
16091609

16101610

1611-
# @torch.jit.script
16121611
def select_ext_factor(
16131612
seq_len: torch.Tensor, max_pos_embeddings: torch.Tensor, short_factor: torch.Tensor, long_factor: torch.Tensor
16141613
):
16151614
return torch.where(
16161615
seq_len < max_pos_embeddings, short_factor, long_factor
1617-
) # short_factor * (seq_len <= max_pos_embeddings) + long_factor * (seq_len > max_pos_embeddings)
1616+
)
16181617

16191618

16201619
def long_rope(self, x, position_ids, seq_len=None):

0 commit comments

Comments
 (0)