ydshieh commited on
Commit
7834fdb
1 Parent(s): bbf2c33
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.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
 
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