diff --git a/docs/source/openvino/models.mdx b/docs/source/openvino/models.mdx index 4e4d34a5b4..84dc0a60d2 100644 --- a/docs/source/openvino/models.mdx +++ b/docs/source/openvino/models.mdx @@ -53,6 +53,7 @@ Here is the list of the supported architectures : - Encoder Decoder - ESM - Exaone +- Exaone4 - Falcon - Falcon-Mamba - Flaubert diff --git a/optimum/exporters/openvino/model_configs.py b/optimum/exporters/openvino/model_configs.py index b8ffcdcf26..f32cc56017 100644 --- a/optimum/exporters/openvino/model_configs.py +++ b/optimum/exporters/openvino/model_configs.py @@ -593,6 +593,21 @@ class ExaoneOpenVINOConfig(LlamaOpenVINOConfig): pass +@register_in_tasks_manager( + "exaone4", + *[ + "feature-extraction", + "feature-extraction-with-past", + "text-generation", + "text-generation-with-past", + "text-classification", + ], + library_name="transformers", +) +class Exaone4OpenVINOConfig(LlamaOpenVINOConfig): + MIN_TRANSFORMERS_VERSION = "4.54.0" + + @register_in_tasks_manager( "arcee", *[ diff --git a/tests/openvino/test_decoder.py b/tests/openvino/test_decoder.py index 007b256a7e..1b40d8029c 100644 --- a/tests/openvino/test_decoder.py +++ b/tests/openvino/test_decoder.py @@ -119,6 +119,7 @@ class OVModelForCausalLMIntegrationTest(unittest.TestCase): if is_transformers_version(">=", "4.54.0"): # remote code models differs after transformers v4.54 + SUPPORTED_ARCHITECTURES += ("exaone4",) SUPPORTED_ARCHITECTURES = tuple(set(SUPPORTED_ARCHITECTURES) - {"minicpm", "minicpm3", "arctic", "deepseek"}) if is_transformers_version(">=", "4.55.0"): @@ -142,6 +143,7 @@ class OVModelForCausalLMIntegrationTest(unittest.TestCase): "arctic", "chatglm4", "exaone", + "exaone4", "decilm", "minicpm3", "deepseek", @@ -196,6 +198,7 @@ class OVModelForCausalLMIntegrationTest(unittest.TestCase): "phi3": 2, "gemma2": 4, "exaone": 8, + "exaone4": 1, "granite": 6, "granite-moe": 6, "glm": 28, diff --git a/tests/openvino/utils_tests.py b/tests/openvino/utils_tests.py index 3a5274a80b..8e7ba9b684 100644 --- a/tests/openvino/utils_tests.py +++ b/tests/openvino/utils_tests.py @@ -78,6 +78,7 @@ "electra": "optimum-intel-internal-testing/tiny-random-electra", "esm": "optimum-intel-internal-testing/tiny-random-EsmModel", "exaone": "optimum-intel-internal-testing/tiny-random-exaone", + "exaone4": "optimum-intel-internal-testing/tiny-random-exaone4", "gemma": "optimum-intel-internal-testing/tiny-random-GemmaForCausalLM", "gemma2": "optimum-intel-internal-testing/tiny-random-gemma2", "got_ocr2": "optimum-intel-internal-testing/tiny-random-got-ocr2-hf",