NielsRogge
|
da36c557f7
|
Add ImageGPT (#14240)
* First draft
* More improvements
* Improve conversion script
* Fix init weights for layer norm
* Fix correct model for conversion script
* Don't tie input and output embeddings
* Add print statements for debugging
* Add print statements for debugging
* Fix vocab size of model
* Improve documentation, remove fast tokenizer
* Add ImageGPTForImageClassification, improve docs
* Fix docs issue
* Set verbosity level back to info
* Improve tests
* Fix tests and add figure
* Delete tokenizer file
* Remove ImageGPTTokenizer from init files
* Remove ImageGPTLayer from init files
* Remove ImageGPT tokenizer from docs
* First draft of ImageGPTFeatureExtractor
* Fix typo
* Fix bug
* More improvements
* Apply suggestions from code review, add tests for feature extractor
* Fix layernorm
* Update save_pretrained method
* Fix issue
* Make all tests of ImageGPTFeatureExtractor pass
* Update code examples
* Rename model inputs to pixel_values
* Improve code examples
* Update init_weights to post_init
* Fix post_init
|
2021-11-18 16:24:34 +01:00 |
|