update setup

huggingface · Oct 10, 2024 · 6443ea3 · 6443ea3
1 parent cb5208c
commit 6443ea3
Show file tree

Hide file tree

Showing 2 changed files with 2 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.44.*"]
+        transformers-version: ["4.36.0", "4.45.*"]
         os: [ubuntu-latest]
 
     runs-on: ${{ matrix.os }}

diff --git a/setup.py b/setup.py
@@ -29,7 +29,7 @@
 INSTALL_REQUIRE = [
     "torch>=1.11",
     "transformers>=4.36,<4.46",
-    "optimum@git+https://github.com/huggingface/optimum.git",
+    "optimum~=1.23",
     "datasets>=1.4.0",
     "sentencepiece",
     "setuptools",