File tree 2 files changed +17
-12
lines changed
2 files changed +17
-12
lines changed Original file line number Diff line number Diff line change 26
26
matrix :
27
27
os : ["ubuntu-22.04", "windows-2019"]
28
28
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"
29
34
30
35
runs-on : ${{ matrix.os }}
31
36
65
70
run : |
66
71
pip install .[nncf]
67
72
68
- - if : ${{ matrix.transformers-version != 'latest' }}
69
- name : Downgrade Transformers and Accelerate
70
- run : pip install transformers==${{ matrix.transformers-version }} accelerate==0.*
71
-
72
73
- name : Test with Pytest (slow)
73
74
run : |
74
75
pytest tests/openvino -m "run_slow" --durations=0
Original file line number Diff line number Diff line change @@ -980,24 +980,28 @@ class OVModelForCausalLMIntegrationTest(unittest.TestCase):
980
980
if is_transformers_version (">=" , "4.40.0" ):
981
981
SUPPORTED_ARCHITECTURES += (
982
982
"gemma" ,
983
- "gemma2" ,
984
983
"olmo" ,
985
984
"stablelm" ,
986
985
"starcoder2" ,
987
986
"dbrx" ,
988
- "phi3" ,
989
987
"cohere" ,
990
988
"qwen2" ,
991
989
"qwen2-moe" ,
992
990
"arctic" ,
993
- "exaone" ,
994
- "mistral-nemo" ,
995
- "minicpm3" ,
996
- "glm" ,
997
- "granite" ,
998
- "granite-moe" ,
999
991
)
1000
992
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
+
1001
1005
# gptq and awq install disabled for windows test environment
1002
1006
if platform .system () != "Windows" :
1003
1007
SUPPORTED_ARCHITECTURES += ("opt_gptq" ,)
You can’t perform that action at this time.
0 commit comments