tmp for test

huggingface · Jul 3, 2024 · 5e9be54 · 5e9be54
1 parent abc1428
commit 5e9be54
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/.github/workflows/test_openvino.yml b/.github/workflows/test_openvino.yml
@@ -21,7 +21,7 @@ jobs:
       fail-fast: false
       matrix:
         python-version: ["3.8", "3.12"]
-        transformers-version: ["4.36.0", "4.42.*"]
+        transformers-version: ["4.36.0", "4.41.*","4.42.*"]
         os: [ubuntu-latest]
 
     runs-on: ${{ matrix.os }}

diff --git a/setup.py b/setup.py
@@ -29,7 +29,8 @@
 INSTALL_REQUIRE = [
     "torch>=1.11",
     "transformers>=4.36.0,<4.43.0",
-    "optimum~=1.21",
+    "optimum @ git+https://github.com/huggingface/optimum.git",
+    #"optimum>=1.21.2,<1.22.0",
     "datasets>=1.4.0",
     "sentencepiece",
     "setuptools",