We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 62f570f commit bbdd34cCopy full SHA for bbdd34c
optimum/intel/ipex/modeling_base.py
@@ -282,7 +282,7 @@ def forward(
282
inputs["attention_mask"] = attention_mask
283
284
outputs = self._call_model(**inputs)
285
- return ModelOutput(**outputs) if isinstance(outputs, dict) else ModelOutput(logits=outputs[0])
+ return ModelOutput(**outputs) if isinstance(outputs, dict) else ModelOutput(last_hidden_state=outputs[0])
286
287
288
class IPEXModelForQuestionAnswering(IPEXModel):
0 commit comments