ydshieh
commited on
Commit
·
e00ec09
1
Parent(s):
49030fb
update 4
Browse files
run_image_captioning_flax_reduced.py
CHANGED
|
@@ -504,7 +504,9 @@ def main():
|
|
| 504 |
"Decoder Config: Either a pretrained config or a model location for decoder is required."
|
| 505 |
)
|
| 506 |
|
| 507 |
-
# GPT2 only has eos token but not pad token
|
|
|
|
|
|
|
| 508 |
if decoder_config.pad_token_id is None:
|
| 509 |
decoder_config.pad_token_id = decoder_config.eos_token_id
|
| 510 |
|
|
|
|
| 504 |
"Decoder Config: Either a pretrained config or a model location for decoder is required."
|
| 505 |
)
|
| 506 |
|
| 507 |
+
# GPT2 only has bos/eos token but not decoder_start/pad token
|
| 508 |
+
if decoder_config.decoder_start_token_id is None:
|
| 509 |
+
decoder_config.decoder_start_token_id = decoder_config.bos_token_id
|
| 510 |
if decoder_config.pad_token_id is None:
|
| 511 |
decoder_config.pad_token_id = decoder_config.eos_token_id
|
| 512 |
|