ydshieh commited on
Commit
ea4daa2
·
1 Parent(s): bcae421
run_image_captioning_flax_reduced.py CHANGED
@@ -526,6 +526,7 @@ def main():
526
  # necessary to make Flax's generate() work
527
  model.config.eos_token_id = decoder_config.eos_token_id
528
  model.config.decoder_start_token_id = decoder_config.decoder_start_token_id
 
529
 
530
  if model_args.feature_extractor_name:
531
  feature_extractor = AutoFeatureExtractor.from_pretrained(
 
526
  # necessary to make Flax's generate() work
527
  model.config.eos_token_id = decoder_config.eos_token_id
528
  model.config.decoder_start_token_id = decoder_config.decoder_start_token_id
529
+ model.config.pad_token_id = decoder_config.pad_token_id
530
 
531
  if model_args.feature_extractor_name:
532
  feature_extractor = AutoFeatureExtractor.from_pretrained(