ydshieh commited on
Commit
af2d49a
·
1 Parent(s): 18ea816
Files changed (1) hide show
  1. run_image_captioning_flax.py +1 -1
run_image_captioning_flax.py CHANGED
@@ -797,7 +797,7 @@ def main():
797
  if k != image_column:
798
  examples[k] = v[bools]
799
  else:
800
- assert len(images) == len(examples)
801
 
802
  encoder_inputs = feature_extractor(images=images, return_tensors="np")
803
  model_inputs = {"pixel_values": encoder_inputs.pixel_values}
 
797
  if k != image_column:
798
  examples[k] = v[bools]
799
  else:
800
+ assert len(images) == len(examples[image_column])
801
 
802
  encoder_inputs = feature_extractor(images=images, return_tensors="np")
803
  model_inputs = {"pixel_values": encoder_inputs.pixel_values}