1. 10 Mar, 2022 1 commit
    • NielsRogge's avatar
      Add Document Image Transformer (DiT) (#15984) · 0835119b
      NielsRogge authored
      
      
      * Add conversion script
      
      * Improve script
      
      * Fix bug
      
      * Add option to push to hub
      
      * Add support for classification models
      
      * Update model name
      
      * Upload feature extractor files first
      
      * Remove hash checking
      
      * Fix config
      
      * Add id2label
      
      * Add import
      
      * Fix id2label file name
      
      * Fix expected shape
      
      * Add model to README
      
      * Improve docs
      
      * Add integration test and fix CI
      
      * Fix code style
      
      * Add missing init
      
      * Add model to SPECIAL_MODULE_TO_TEST_MAP
      Co-authored-by: default avatarNiels Rogge <nielsrogge@Nielss-MacBook-Pro.local>
      0835119b