Skip to content

Commit bbdd34c

Browse files
committed
change model output parameter to last_hidden_states
1 parent 62f570f commit bbdd34c

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

optimum/intel/ipex/modeling_base.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -282,7 +282,7 @@ def forward(
282282
inputs["attention_mask"] = attention_mask
283283

284284
outputs = self._call_model(**inputs)
285-
return ModelOutput(**outputs) if isinstance(outputs, dict) else ModelOutput(logits=outputs[0])
285+
return ModelOutput(**outputs) if isinstance(outputs, dict) else ModelOutput(last_hidden_state=outputs[0])
286286

287287

288288
class IPEXModelForQuestionAnswering(IPEXModel):

0 commit comments

Comments
 (0)