add granite testwhen supported by transformers

huggingface · Nov 18, 2024 · 82bfbbf · 82bfbbf
1 parent 3c1f85f
commit 82bfbbf
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/tests/onnxruntime/test_modeling.py b/tests/onnxruntime/test_modeling.py
@@ -2323,7 +2323,6 @@ class ORTModelForCausalLMIntegrationTest(ORTModelTestMixin):
         "gpt_neo",
         "gpt_neox",
         "gptj",
-        "granite",
         "llama",
         "mistral",
         "opt",
@@ -2339,6 +2338,9 @@ class ORTModelForCausalLMIntegrationTest(ORTModelTestMixin):
     if check_if_transformers_greater("4.41"):
         SUPPORTED_ARCHITECTURES.extend(["phi3", "mpt"])
 
+    if check_if_transformers_greater("4.45"):
+        SUPPORTED_ARCHITECTURES.append("granite")
+
     FULL_GRID = {
         "model_arch": SUPPORTED_ARCHITECTURES,
         "use_cache": [False, True],