Bump transformers version

echarlaix · echarlaix · commit ebca14dfaae5 · 2024-03-21T11:38:01.000+01:00
diff --git a/setup.py b/setup.py
@@ -28,7 +28,7 @@
 
 INSTALL_REQUIRE = [
     "torch>=1.11",
-    "transformers>=4.36.0,<4.39.0",
+    "transformers>=4.36.0,<4.40.0",
     "optimum @ git+https://github.com/huggingface/optimum.git#egg=optimum",
     "datasets>=1.4.0",
     "sentencepiece",
diff --git a/tests/openvino/test_modeling.py b/tests/openvino/test_modeling.py
@@ -501,7 +501,7 @@ class OVModelForCausalLMIntegrationTest(unittest.TestCase):
         "qwen",
         "qwen2",
         "stablelm",
-        # "starcoder2", # TODO: enable with next transformers release
+        "starcoder2",
     )
     GENERATION_LENGTH = 100
     IS_SUPPORT_STATEFUL = is_openvino_version(">=", "2023.3")

Original file line number	Diff line number	Diff line change
`@@ -501,7 +501,7 @@ class OVModelForCausalLMIntegrationTest(unittest.TestCase):`
`501`	`501`	`"qwen",`
`502`	`502`	`"qwen2",`
`503`	`503`	`"stablelm",`
`504`		`- # "starcoder2", # TODO: enable with next transformers release`
	`504`	`+ "starcoder2",`
`505`	`505`	`)`
`506`	`506`	`GENERATION_LENGTH = 100`
`507`	`507`	`IS_SUPPORT_STATEFUL = is_openvino_version(">=", "2023.3")`