ydshieh
/

flax-vision-encoder-decoder-vit-gpt2-coco-en

Model card Files Files and versions

ydshieh commited on Dec 18, 2021

Commit

bcdd5f7

·

1 Parent(s): b05f10c

improve doc

Files changed (1) hide show

run_image_captioning_flax.py +1 -0

run_image_captioning_flax.py CHANGED Viewed

@@ -803,6 +803,7 @@ def main():
         """
         model_inputs = {}
         model_inputs.update(tokenization_fn(examples, max_target_length))
         model_inputs.update(feature_extraction_fn(model_inputs))
         # Remove image path column

         """
         model_inputs = {}
+        # This contains image path column
         model_inputs.update(tokenization_fn(examples, max_target_length))
         model_inputs.update(feature_extraction_fn(model_inputs))
         # Remove image path column