ydshieh
/

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

Model card Files Files and versions

ydshieh commited on Dec 19, 2021

Commit

65ed6f6

·

1 Parent(s): 0913b91

fix

Files changed (1) hide show

run_image_captioning_flax.py +1 -1

run_image_captioning_flax.py CHANGED Viewed

@@ -996,7 +996,7 @@ def main():
                     features=features,
                     keep_in_memory=keep_in_memory,
                     # The images are already checked either in `.filter()` or in `preprocess_fn()`
-                    fn_kwargs={"check": False},
                     desc=f"Running feature extraction on {split} dataset".replace("  ", " "),
                 )
                 _ds = _ds.with_format("numpy")

                     features=features,
                     keep_in_memory=keep_in_memory,
                     # The images are already checked either in `.filter()` or in `preprocess_fn()`
+                    fn_kwargs={"check_image": False},
                     desc=f"Running feature extraction on {split} dataset".replace("  ", " "),
                 )
                 _ds = _ds.with_format("numpy")