Skip to content

Commit efedca4

Browse files
committed
revert openvino
1 parent 68187e5 commit efedca4

File tree

1 file changed

+3
-5
lines changed

1 file changed

+3
-5
lines changed

optimum/intel/openvino/modeling_decoder.py

+3-5
Original file line numberDiff line numberDiff line change
@@ -571,11 +571,9 @@ def _expand_outputs_for_generation(self, indicies, logits: torch.Tensor, past_ke
571571
):
572572
past_key_values = tuple(
573573
tuple(
574-
(
575-
past_state[indicies]
576-
if not self.config.model_type == "chatglm"
577-
else past_state[:, indicies, ...]
578-
)
574+
past_state[indicies]
575+
if not self.config.model_type == "chatglm"
576+
else past_state[:, indicies, ...]
579577
for past_state in layer_past
580578
)
581579
for layer_past in past_key_values

0 commit comments

Comments
 (0)