ydshieh
commited on
Commit
·
af2d49a
1
Parent(s):
18ea816
fix
Browse files
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}
|