From 3ff89f29f542dc0e16b48e7b8d7710555f454d5c Mon Sep 17 00:00:00 2001 From: "Gregory (Gabriel) Barello" <48561156+gbarello-uipath@users.noreply.github.com> Date: Tue, 2 May 2023 10:40:41 -0700 Subject: [PATCH] Fixed default config for `Pix2Struct` model to set `Pix2StructTextModel` to `is_decoder=True` (#23051) added as default keyword arg. to in order to correctly configure the decoder --- src/transformers/models/pix2struct/configuration_pix2struct.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/src/transformers/models/pix2struct/configuration_pix2struct.py b/src/transformers/models/pix2struct/configuration_pix2struct.py index 244cb270586..32aa34941f8 100644 --- a/src/transformers/models/pix2struct/configuration_pix2struct.py +++ b/src/transformers/models/pix2struct/configuration_pix2struct.py @@ -118,6 +118,7 @@ class Pix2StructTextConfig(PretrainedConfig): pad_token_id=0, eos_token_id=1, tie_word_embeddings=False, + is_decoder=True, **kwargs, ): self.vocab_size = vocab_size @@ -144,6 +145,7 @@ class Pix2StructTextConfig(PretrainedConfig): eos_token_id=eos_token_id, decoder_start_token_id=decoder_start_token_id, tie_word_embeddings=tie_word_embeddings, + is_decoder=is_decoder, **kwargs, )