Skip to content
Merged
Show file tree
Hide file tree
Changes from 1 commit
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 @@ -17,6 +17,7 @@ Here is the list of the supported architectures :
- Albert
- Aquila
- Aquila 2
- Arcee
- Arctic
- Audio Spectrogram Transformer
- Baichuan 2
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 @@ -617,6 +617,21 @@ class ExaoneOpenVINOConfig(LlamaOpenVINOConfig):
pass


@register_in_tasks_manager(
"arcee",
*[
"feature-extraction",
"feature-extraction-with-past",
"text-generation",
"text-generation-with-past",
"text-classification",
],
library_name="transformers",
)
class ArceeOpenVINOConfig(LlamaOpenVINOConfig):
MIN_TRANSFORMERS_VERSION = "4.53.0"


class QwenDummyPastKeyValuesGenerator(DummyPastKeyValuesGenerator):
def __init__(
self,
Expand Down
4 changes: 4 additions & 0 deletions tests/openvino/test_modeling.py
Original file line number Diff line number Diff line change
Expand Up @@ -1183,6 +1183,9 @@ class OVModelForCausalLMIntegrationTest(unittest.TestCase):
if is_transformers_version(">=", "4.51.3"):
SUPPORTED_ARCHITECTURES += ("glm4",)

if is_transformers_version(">=", "4.53.0"):
SUPPORTED_ARCHITECTURES += ("arcee",)

GENERATION_LENGTH = 100
REMOTE_CODE_MODELS = (
"chatglm",
Expand Down Expand Up @@ -1270,6 +1273,7 @@ class OVModelForCausalLMIntegrationTest(unittest.TestCase):
"qwen3_moe": 2,
"mamba": 0,
"falcon-mamba": 0,
"arcee": 2,
}

# TODO: remove gptq/awq from here
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 @@ -27,6 +27,7 @@
"albert": "hf-internal-testing/tiny-random-albert",
"aquila": "katuni4ka/tiny-random-aquilachat",
"aquila2": "katuni4ka/tiny-random-aquila2",
"arcee": "onnx-internal-testing/tiny-random-ArceeForCausalLM",
"arctic": "katuni4ka/tiny-random-snowflake",
"audio-spectrogram-transformer": "Ericwang/tiny-random-ast",
"bge": "BAAI/bge-small-en-v1.5",
Expand Down
Loading