diff --git a/docs/source/openvino/models.mdx b/docs/source/openvino/models.mdx index db2701749f..0be3a8692b 100644 --- a/docs/source/openvino/models.mdx +++ b/docs/source/openvino/models.mdx @@ -116,6 +116,7 @@ Here is the list of the supported architectures : - MobileNet v1 - MobileNet v2 - MobileViT +- Nomic Bert - Nystromformer - OLMo - OLMo 2 diff --git a/optimum/exporters/openvino/model_configs.py b/optimum/exporters/openvino/model_configs.py index fcacba0534..903cd305d2 100644 --- a/optimum/exporters/openvino/model_configs.py +++ b/optimum/exporters/openvino/model_configs.py @@ -5576,6 +5576,11 @@ class MobileBertOpenVINOConfig(BertOpenVINOConfig): pass +@register_in_tasks_manager("nomic_bert", *COMMON_TEXT_TASKS) +class NomicBertOpenVINOConfig(BertOpenVINOConfig): + pass + + @register_in_tasks_manager("xlm", *COMMON_TEXT_TASKS) class XLMOpenVINOConfig(BertOpenVINOConfig): MAX_TRANSFORMERS_VERSION = "4.57.6" diff --git a/tests/openvino/test_modeling.py b/tests/openvino/test_modeling.py index 196d22a6a7..a24ed45700 100644 --- a/tests/openvino/test_modeling.py +++ b/tests/openvino/test_modeling.py @@ -1037,6 +1037,7 @@ class OVModelForFeatureExtractionIntegrationTest(unittest.TestCase): SUPPORTED_ARCHITECTURES = ( "bert", "distilbert", + "nomic_bert", "roberta", "sentence-transformers-bert", "qwen3", diff --git a/tests/openvino/utils_tests.py b/tests/openvino/utils_tests.py index f0e32b265b..75798c449b 100644 --- a/tests/openvino/utils_tests.py +++ b/tests/openvino/utils_tests.py @@ -276,6 +276,7 @@ def _create_tiny_kokoro_model(): "mt5": "optimum-intel-internal-testing/mt5-tiny-random", "llava-qwen2": "optimum-intel-internal-testing/tiny-random-nanollava", "nanollava_vision_tower": "optimum-intel-internal-testing/tiny-random-siglip", + "nomic_bert": "optimum-intel-internal-testing/tiny-random-nomic-bert", "nystromformer": "optimum-intel-internal-testing/tiny-random-NystromformerModel", "olmo": "optimum-intel-internal-testing/tiny-random-olmo-hf", "orion": "optimum-intel-internal-testing/tiny-random-orion",