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