"git@developer.sourcefind.cn:change/sglang.git" did not exist on "93414c8238c0fae97b8c741940f33dff58aec7c6"
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): ...@@ -118,6 +118,7 @@ class Pix2StructTextConfig(PretrainedConfig):
pad_token_id=0, pad_token_id=0,
eos_token_id=1, eos_token_id=1,
tie_word_embeddings=False, tie_word_embeddings=False,
is_decoder=True,
**kwargs, **kwargs,
): ):
self.vocab_size = vocab_size self.vocab_size = vocab_size
...@@ -144,6 +145,7 @@ class Pix2StructTextConfig(PretrainedConfig): ...@@ -144,6 +145,7 @@ class Pix2StructTextConfig(PretrainedConfig):
eos_token_id=eos_token_id, eos_token_id=eos_token_id,
decoder_start_token_id=decoder_start_token_id, decoder_start_token_id=decoder_start_token_id,
tie_word_embeddings=tie_word_embeddings, tie_word_embeddings=tie_word_embeddings,
is_decoder=is_decoder,
**kwargs, **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