ydshieh
commited on
Commit
•
7834fdb
1
Parent(s):
bbf2c33
update 13
Browse files
run_image_captioning_flax_reduced.py
CHANGED
@@ -524,7 +524,7 @@ def main():
|
|
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.
|
528 |
|
529 |
# This is necessary to make Flax's generate() work
|
530 |
model.config.eos_token_id = decoder_config.eos_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.eos_token_id
|
528 |
|
529 |
# This is necessary to make Flax's generate() work
|
530 |
model.config.eos_token_id = decoder_config.eos_token_id
|