Skip to content

Commit 913b4b5

Browse files
authored
Add slow tests for lower transformers version (huggingface#1144)
* Add slow tests for lower transformers version * fix tests * remove phi3 * trigger test
1 parent 9898189 commit 913b4b5

File tree

2 files changed

+17
-12
lines changed

2 files changed

+17
-12
lines changed

.github/workflows/test_openvino_slow.yml

+5-4
Original file line numberDiff line numberDiff line change
@@ -26,6 +26,11 @@ jobs:
2626
matrix:
2727
os: ["ubuntu-22.04", "windows-2019"]
2828
transformers-version: ["4.36.0", "latest"]
29+
include:
30+
- transformers-version: "4.40.0"
31+
os: "ubuntu-22.04"
32+
- transformers-version: "4.45.0"
33+
os: "ubuntu-22.04"
2934

3035
runs-on: ${{ matrix.os }}
3136

@@ -65,10 +70,6 @@ jobs:
6570
run: |
6671
pip install .[nncf]
6772
68-
- if: ${{ matrix.transformers-version != 'latest' }}
69-
name: Downgrade Transformers and Accelerate
70-
run: pip install transformers==${{ matrix.transformers-version }} accelerate==0.*
71-
7273
- name: Test with Pytest (slow)
7374
run: |
7475
pytest tests/openvino -m "run_slow" --durations=0

tests/openvino/test_modeling.py

+12-8
Original file line numberDiff line numberDiff line change
@@ -980,24 +980,28 @@ class OVModelForCausalLMIntegrationTest(unittest.TestCase):
980980
if is_transformers_version(">=", "4.40.0"):
981981
SUPPORTED_ARCHITECTURES += (
982982
"gemma",
983-
"gemma2",
984983
"olmo",
985984
"stablelm",
986985
"starcoder2",
987986
"dbrx",
988-
"phi3",
989987
"cohere",
990988
"qwen2",
991989
"qwen2-moe",
992990
"arctic",
993-
"exaone",
994-
"mistral-nemo",
995-
"minicpm3",
996-
"glm",
997-
"granite",
998-
"granite-moe",
999991
)
1000992

993+
if is_transformers_version(">=", "4.41.0"):
994+
SUPPORTED_ARCHITECTURES += ("phi3",)
995+
996+
if is_transformers_version(">=", "4.43.0"):
997+
SUPPORTED_ARCHITECTURES += ("gemma2", "exaone")
998+
999+
if is_transformers_version(">=", "4.44.0"):
1000+
SUPPORTED_ARCHITECTURES += ("granite", "granite-moe")
1001+
1002+
if is_transformers_version(">=", "4.46.0"):
1003+
SUPPORTED_ARCHITECTURES += ("glm", "mistral-nemo", "minicpm3")
1004+
10011005
# gptq and awq install disabled for windows test environment
10021006
if platform.system() != "Windows":
10031007
SUPPORTED_ARCHITECTURES += ("opt_gptq",)

0 commit comments

Comments
 (0)