Unverified Commit 3ff89f29 authored by Gregory (Gabriel) Barello's avatar Gregory (Gabriel) Barello Committed by GitHub
Browse files

Fixed default config for `Pix2Struct` model to set `Pix2StructTextModel` to...

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
parent 805db1fe
......@@ -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,
)
......
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