Skip to content

Commit

Permalink
typo
Browse files Browse the repository at this point in the history
  • Loading branch information
IlyasMoutawwakil committed Jan 15, 2025
1 parent 88a7e8b commit e9abe6a
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions optimum/onnxruntime/modeling_seq2seq.py
Original file line number Diff line number Diff line change
Expand Up @@ -433,7 +433,7 @@ def forward(
else:
onnx_inputs = self._prepare_onnx_inputs(use_torch, **model_inputs)
onnx_outputs = self.session.run(None, onnx_inputs)
model_outputs = self._prepare_model_outputs(use_torch, *onnx_outputs)
model_outputs = self._prepare_onnx_outputs(use_torch, *onnx_outputs)

last_hidden_state = model_outputs["last_hidden_state"]

Expand Down Expand Up @@ -475,7 +475,7 @@ def forward(
else:
onnx_inputs = self._prepare_onnx_inputs(use_torch, **model_inputs)
onnx_outputs = self.session.run(None, onnx_inputs)
model_outputs = self._prepare_model_outputs(use_torch, *onnx_outputs)
model_outputs = self._prepare_onnx_outputs(use_torch, *onnx_outputs)

last_hidden_state = model_outputs["last_hidden_state"]

Expand Down

0 comments on commit e9abe6a

Please sign in to comment.