1. 23 Feb, 2022 1 commit
    • Nicolas Patry's avatar
      Adding ZeroShotImageClassificationPipeline (#12119) · f9582c20
      Nicolas Patry authored
      
      
      * [Proposal] Adding ZeroShotImageClassificationPipeline
      
      - Based on CLIP
      
      * WIP, Resurection in progress.
      
      * Resurrection... achieved.
      
      * Reword handling different `padding_value` for `feature_extractor` and
      `tokenizer`.
      
      * Thanks doc-builder !
      
      * Adding docs + global namespace `ZeroShotImageClassificationPipeline`.
      
      * Fixing templates.
      
      * Make the test pass and be robust to floating error.
      
      * Adressing suraj's comments on docs mostly.
      
      * Tf support start.
      
      * TF support.
      
      * Update src/transformers/pipelines/zero_shot_image_classification.py
      Co-authored-by: default avatarSuraj Patil <surajp815@gmail.com>
      Co-authored-by: default avatarSuraj Patil <surajp815@gmail.com>
      f9582c20