Skip to content
Open
Show file tree
Hide file tree
Changes from 8 commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
1 change: 1 addition & 0 deletions docs/source/openvino/models.mdx
Original file line number Diff line number Diff line change
Expand Up @@ -53,6 +53,7 @@ Here is the list of the supported architectures :
- Encoder Decoder
- ESM
- Exaone
- Exaone4
- Falcon
- Falcon-Mamba
- Flaubert
Expand Down
15 changes: 15 additions & 0 deletions optimum/exporters/openvino/model_configs.py
Original file line number Diff line number Diff line change
Expand Up @@ -597,6 +597,21 @@ class ExaoneOpenVINOConfig(LlamaOpenVINOConfig):
pass


@register_in_tasks_manager(
"exaone4",
*[
"feature-extraction",
"feature-extraction-with-past",
"text-generation",
"text-generation-with-past",
"text-classification",
Copy link
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

OVModelForSequenceClassification is not validated. I see that this task is not validated in precommit.
The same problem is for feature-extraction.

],
library_name="transformers",
)
class Exaone4OpenVINOConfig(LlamaOpenVINOConfig):
MIN_TRANSFORMERS_VERSION = "4.54.0"


@register_in_tasks_manager(
"arcee",
*[
Expand Down
3 changes: 3 additions & 0 deletions tests/openvino/test_decoder.py
Original file line number Diff line number Diff line change
Expand Up @@ -122,6 +122,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"):
Expand All @@ -145,6 +146,7 @@ class OVModelForCausalLMIntegrationTest(unittest.TestCase):
"arctic",
"chatglm4",
"exaone",
"exaone4",
"decilm",
"minicpm3",
"deepseek",
Expand Down Expand Up @@ -199,6 +201,7 @@ class OVModelForCausalLMIntegrationTest(unittest.TestCase):
"phi3": 2,
"gemma2": 4,
"exaone": 8,
"exaone4": 1,
"granite": 6,
"granite-moe": 6,
"glm": 28,
Expand Down
2 changes: 2 additions & 0 deletions tests/openvino/test_exporters_cli.py
Original file line number Diff line number Diff line change
Expand Up @@ -85,6 +85,7 @@ class OVCLIExportTestCase(unittest.TestCase):
("text2text-generation-with-past", "t5"),
("text-generation-with-past", "mamba"),
("text-generation-with-past", "falcon-mamba"),
("text-generation-with-past", "exaone4"),
("text-classification", "albert"),
("question-answering", "distilbert"),
("token-classification", "roberta"),
Expand Down Expand Up @@ -136,6 +137,7 @@ class OVCLIExportTestCase(unittest.TestCase):
"falcon-mamba": 2,
"qwen3": 2,
"zamba2": 2,
"exaone4": 2,
}

TOKENIZER_CHAT_TEMPLATE_TESTS_MODELS = {
Expand Down
1 change: 1 addition & 0 deletions tests/openvino/utils_tests.py
Original file line number Diff line number Diff line change
Expand Up @@ -77,6 +77,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",
Expand Down
Loading