Skip to content
Open
Show file tree
Hide file tree
Changes from 2 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
14 changes: 14 additions & 0 deletions optimum/exporters/openvino/model_configs.py
Original file line number Diff line number Diff line change
Expand Up @@ -592,6 +592,20 @@ class GptOssOpenVINOConfig(LlamaOpenVINOConfig):
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
Loading