ydshieh
commited on
Commit
·
bbf2c33
1
Parent(s):
ddad56c
update 12
Browse files
run_image_captioning_flax_reduced.py
CHANGED
|
@@ -522,10 +522,10 @@ def main():
|
|
| 522 |
decoder_start_token_id = decoder_config.decoder_start_token_id
|
| 523 |
pad_token_id = decoder_config.pad_token_id
|
| 524 |
if decoder_start_token_id is None:
|
| 525 |
-
|
| 526 |
if pad_token_id is None:
|
| 527 |
pad_token_id = decoder_config.pad_token_id
|
| 528 |
-
|
| 529 |
# This is necessary to make Flax's generate() work
|
| 530 |
model.config.eos_token_id = decoder_config.eos_token_id
|
| 531 |
model.config.decoder_start_token_id = decoder_start_token_id
|
|
|
|
| 522 |
decoder_start_token_id = decoder_config.decoder_start_token_id
|
| 523 |
pad_token_id = decoder_config.pad_token_id
|
| 524 |
if decoder_start_token_id is None:
|
| 525 |
+
decoder_start_token_id = decoder_config.bos_token_id
|
| 526 |
if pad_token_id is None:
|
| 527 |
pad_token_id = decoder_config.pad_token_id
|
| 528 |
+
|
| 529 |
# This is necessary to make Flax's generate() work
|
| 530 |
model.config.eos_token_id = decoder_config.eos_token_id
|
| 531 |
model.config.decoder_start_token_id = decoder_start_token_id
|