Skip to content

Commit 08e3008

Browse files
authored
Fix ESM models export and add it to supported (#1328)
1 parent 902e145 commit 08e3008

File tree

4 files changed

+4
-0
lines changed

4 files changed

+4
-0
lines changed

docs/source/openvino/models.mdx

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -49,6 +49,7 @@ Here is the list of the supported architectures :
4949
- DistilBert
5050
- Electra
5151
- Encoder Decoder
52+
- ESM
5253
- Exaone
5354
- Falcon
5455
- Flaubert

optimum/exporters/openvino/utils.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -369,6 +369,7 @@ def set_simplified_chat_template(ov_tokenizer_model, processor_chat_template=Non
369369
"deepseek",
370370
"deepseek-v2",
371371
"deepseek-v3",
372+
"esm",
372373
"levit",
373374
"llama4",
374375
)

tests/openvino/test_modeling.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1740,6 +1740,7 @@ class OVModelForMaskedLMIntegrationTest(unittest.TestCase):
17401740
"deberta_v2",
17411741
"distilbert",
17421742
"electra",
1743+
"esm",
17431744
"flaubert",
17441745
"ibert",
17451746
"mobilebert",

tests/openvino/utils_tests.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -63,6 +63,7 @@
6363
"donut-swin": "hf-internal-testing/tiny-random-DonutSwinModel",
6464
"detr": "hf-internal-testing/tiny-random-DetrModel",
6565
"electra": "hf-internal-testing/tiny-random-electra",
66+
"esm": "hf-internal-testing/tiny-random-EsmModel",
6667
"exaone": "katuni4ka/tiny-random-exaone",
6768
"gemma": "fxmarty/tiny-random-GemmaForCausalLM",
6869
"gemma2": "katuni4ka/tiny-random-gemma2",

0 commit comments

Comments
 (0)