From 82bfbbfc310a0ac02d558d3ca99b9cc7a8ddb9ae Mon Sep 17 00:00:00 2001
From: Ella Charlaix <ella@huggingface.co>
Date: Mon, 18 Nov 2024 11:42:46 +0100
Subject: [PATCH] add granite testwhen supported by transformers

---
 tests/onnxruntime/test_modeling.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/tests/onnxruntime/test_modeling.py b/tests/onnxruntime/test_modeling.py
index 9aacdd7d96..84ac27029f 100644
--- 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],