Update tests/openvino/test_modeling.py

Co-authored-by: Ella Charlaix <80481427+echarlaix@users.noreply.github.com>
huggingface · Apr 25, 2024 · 84c39aa · 84c39aa
1 parent c7a6d51
commit 84c39aa
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/openvino/test_modeling.py b/tests/openvino/test_modeling.py
@@ -528,7 +528,7 @@ class OVModelForCausalLMIntegrationTest(unittest.TestCase):
         "mistral",
         "mixtral",
         "mpt",
-        # "olmo", # TODO : enable after optimum update for support transformers 4.40
+        "olmo",
         "opt",
         "pegasus",
         "qwen",