ydshieh
/

flax-vision-encoder-decoder-vit-gpt2-coco-en

ydshieh commited on Dec 25, 2021

Commit

bcae421

1 Parent(s): 15ecbe8

update 9

Files changed (1) hide show

run_image_captioning_flax_reduced.py CHANGED Viewed

@@ -524,6 +524,7 @@ def main():
         decoder_dtype=getattr(jnp, model_args.dtype),
     )
     # necessary to make Flax's generate() work
     model.config.decoder_start_token_id = decoder_config.decoder_start_token_id
     if model_args.feature_extractor_name:

         decoder_dtype=getattr(jnp, model_args.dtype),
     )
     # necessary to make Flax's generate() work
+    model.config.eos_token_id = decoder_config.eos_token_id
     model.config.decoder_start_token_id = decoder_config.decoder_start_token_id
     if model_args.feature_extractor_name: