Unverified Commit 9a35a7b9 authored by fxmarty's avatar fxmarty Committed by GitHub
Browse files

Effectively allow `encoder_outputs` input to be a tuple in pix2struct (#23932)

consistentcy
parent 9603ef89
......@@ -1734,6 +1734,12 @@ class Pix2StructForConditionalGeneration(Pix2StructPreTrainedModel):
output_hidden_states=output_hidden_states,
return_dict=return_dict,
)
elif return_dict and not isinstance(encoder_outputs, BaseModelOutput):
encoder_outputs = BaseModelOutput(
last_hidden_state=encoder_outputs[0],
hidden_states=encoder_outputs[1] if len(encoder_outputs) > 1 else None,
attentions=encoder_outputs[2] if len(encoder_outputs) > 2 else None,
)
hidden_states = encoder_outputs[0]
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment