diff --git a/docs/source/exporters/onnx/overview.mdx b/docs/source/exporters/onnx/overview.mdx index 57005b85678..84f17acf4bb 100644 --- a/docs/source/exporters/onnx/overview.mdx +++ b/docs/source/exporters/onnx/overview.mdx @@ -54,6 +54,7 @@ Supported architectures from [🤗 Transformers](https://huggingface.co/docs/tra - OPT - GroupVit - Hubert +- I-JEPA - IBert - LayoutLM - LayoutLM-v3 diff --git a/optimum/exporters/onnx/model_configs.py b/optimum/exporters/onnx/model_configs.py index b39d19ec782..d70dec892ea 100644 --- a/optimum/exporters/onnx/model_configs.py +++ b/optimum/exporters/onnx/model_configs.py @@ -913,6 +913,10 @@ class MobileNetV2OnnxConfig(MobileNetV1OnnxConfig): pass +class IJepaOnnxConfig(ViTOnnxConfig): + ATOL_FOR_VALIDATION = 1e-2 + + class DonutSwinOnnxConfig(ViTOnnxConfig): DEFAULT_ONNX_OPSET = 11 diff --git a/optimum/exporters/tasks.py b/optimum/exporters/tasks.py index 0a3758e97cf..cc9421debf3 100644 --- a/optimum/exporters/tasks.py +++ b/optimum/exporters/tasks.py @@ -747,6 +747,10 @@ class TasksManager: "question-answering", onnx="IBertOnnxConfig", ), + "ijepa": supported_tasks_mapping( + "feature-extraction", + onnx="IJepaOnnxConfig", + ), "imagegpt": supported_tasks_mapping( "feature-extraction", "image-classification",