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:
@@ -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"),
|
||||
|
||||
Reference in New Issue
Block a user