add architectures depending on transformers

huggingface · Oct 29, 2024 · 641d351 · 641d351
1 parent 1ce1a0f
commit 641d351
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/tests/onnxruntime/test_modeling.py b/tests/onnxruntime/test_modeling.py
@@ -107,7 +107,7 @@
     DIFFUSION_MODEL_VAE_ENCODER_SUBFOLDER,
     logging,
 )
-from optimum.utils.import_utils import is_diffusers_available
+from optimum.utils.import_utils import is_diffusers_available, check_if_transformers_greater
 from optimum.utils.testing_utils import (
     grid_parameters,
     remove_directory,
@@ -2316,10 +2316,12 @@ class ORTModelForCausalLMIntegrationTest(ORTModelTestMixin):
         "mistral",
         "mpt",
         "opt",
-        "phi3",
-        "qwen2",
     ]
 
+
+    if check_if_transformers_greater("4.40"):
+        SUPPORTED_ARCHITECTURES.extend(["gemma", "phi3", "qwen2"])
+
     FULL_GRID = {
         "model_arch": SUPPORTED_ARCHITECTURES,
         "use_cache": [False, True],