Skip to content

Commit 7702d35

Browse files
authored
Add tests for latest transformers release (#939)
* Add tests for latest transformers release * style
1 parent b31524c commit 7702d35

File tree

2 files changed

+5
-2
lines changed

2 files changed

+5
-2
lines changed

.github/workflows/test_openvino.yml

+1-1
Original file line numberDiff line numberDiff line change
@@ -21,7 +21,7 @@ jobs:
2121
fail-fast: false
2222
matrix:
2323
python-version: ["3.8", "3.12"]
24-
transformers-version: ["4.36.0", "4.44.*"]
24+
transformers-version: ["4.36.0", "4.45.*"]
2525
os: [ubuntu-latest]
2626

2727
runs-on: ${{ matrix.os }}

tests/openvino/test_modeling.py

+4-1
Original file line numberDiff line numberDiff line change
@@ -771,7 +771,6 @@ class OVModelForCausalLMIntegrationTest(unittest.TestCase):
771771
"bloom",
772772
"chatglm",
773773
"codegen",
774-
"codegen2",
775774
"gpt2",
776775
"gpt_neo",
777776
"gpt_neox",
@@ -820,6 +819,10 @@ class OVModelForCausalLMIntegrationTest(unittest.TestCase):
820819
"mistral-nemo",
821820
)
822821

822+
# custom modeling defined in https://huggingface.co/katuni4ka/tiny-random-codegen2 differs from transformers after v4.45 resulting in unadapted patching
823+
if is_transformers_version("<", "4.45.0"):
824+
SUPPORTED_ARCHITECTURES += ("codegen2",)
825+
823826
GENERATION_LENGTH = 100
824827
REMOTE_CODE_MODELS = (
825828
"chatglm",

0 commit comments

Comments
 (0)