ydshieh
commited on
Commit
·
ea4daa2
1
Parent(s):
bcae421
update 10
Browse files
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(
|