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
|