NielsRogge
|
0835119bf3
|
Add Document Image Transformer (DiT) (#15984)
* 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: Niels Rogge <nielsrogge@Nielss-MacBook-Pro.local>
|
2022-03-10 11:34:44 +01:00 |
|