"tests/clip/test_processor_clip.py" did not exist on "8719afa1adc004011a34a34e374b819b5963f23b"
  • Sanchit Gandhi's avatar
    [Flax] Fix incomplete batches in example scripts (#17863) · 7490a97c
    Sanchit Gandhi authored
    * [Flax] Fix incomplete batches in example scripts
    
    * fix dataloader batching
    
    * convert jnp batch idxs to np array
    
    * add missing `pad_shard_unpad` to final prediction generate step
    
    * only `pad_shard_unpad` at inference time
    
    * merge conflicts
    
    * remove incomplete batch step from eval
    
    * fix run_qa.py
    
    * add `pad_shard_unpad` to run_flax_ner.py
    
    * add `pad_shard_unpad` to run_flax_glue.py
    
    * add `pad_shard_unpad` to run_image_classification.py
    
    * make style
    
    * fix mlm flax eval batches
    
    * remove redundant imports
    7490a97c
run_summarization_flax.py 40.5 KB