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(
|