Add Optimum OV support for Arcee AFM (#1401)

Signed-off-by: Mateusz Mikolajczyk <[email protected]>

Author: mmikolajcz

docs/source/openvino/models.mdx

@@ -17,6 +17,7 @@ Here is the list of the supported architectures :
 - Albert
 - Aquila
 - Aquila 2
+- Arcee
 - Arctic
 - Audio Spectrogram Transformer
 - Baichuan 2

optimum/exporters/openvino/model_configs.py

@@ -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,

tests/openvino/test_modeling.py

@@ -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",)
+
     if is_transformers_version(">=", "4.54.0"):
         SUPPORTED_ARCHITECTURES += ("ernie4_5",)
 
@@ -1273,6 +1276,7 @@ class OVModelForCausalLMIntegrationTest(unittest.TestCase):
         "qwen3_moe": 2,
         "mamba": 0,
         "falcon-mamba": 0,
+        "arcee": 2,
         "ernie4_5": 2,
     }
 

tests/openvino/utils_tests.py

@@ -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",