Add Onnx Config for ImageGPT (#19868)

* add Onnx Config for ImageGPT

* add generate_dummy_inputs for onnx config

* add TYPE_CHECKING clause

* Update doc for generate_dummy_inputs

Co-authored-by: Sylvain Gugger <35901082+sgugger@users.noreply.github.com>

Co-authored-by: Sylvain Gugger <35901082+sgugger@users.noreply.github.com>
This commit is contained in:
Raghav Prabhakar
2022-10-28 19:09:53 +05:30
committed by GitHub
parent 9b1dcba94a
commit 0d4c45c585
5 changed files with 69 additions and 2 deletions

View File

@@ -193,6 +193,7 @@ PYTORCH_EXPORT_MODELS = {
("detr", "facebook/detr-resnet-50"),
("distilbert", "distilbert-base-cased"),
("electra", "google/electra-base-generator"),
("imagegpt", "openai/imagegpt-small"),
("resnet", "microsoft/resnet-50"),
("roberta", "roberta-base"),
("roformer", "junnyu/roformer_chinese_base"),