• 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
_toctree.yml 10.4 KB