fix

huggingface · Oct 27, 2024 · 4305232 · 4305232
1 parent 8993575
commit 4305232
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/.github/workflows/test_generation.yml b/.github/workflows/test_generation.yml
@@ -1,4 +1,4 @@
-name: Generation Utils - Test
+name: Generation Utils - Test (deprecated)
 
 on:
   push:

diff --git a/.github/workflows/test_ipex.yml b/.github/workflows/test_ipex.yml
@@ -40,7 +40,7 @@ jobs:
           pip install torch==${{ matrix.torch-version }} torchaudio torchvision --extra-index-url https://download.pytorch.org/whl/cpu
           pip install intel_extension_for_pytorch==${{ matrix.torch-version }}
           pip install transformers==${{ matrix.transformers-version }}
-          pip install .[ipex]
+          pip install .[ipex,tests]
 
       - name: Assert versions
         run: |

diff --git a/setup.py b/setup.py
@@ -28,7 +28,7 @@
 
 INSTALL_REQUIRE = [
     "torch>=1.11",
-    "transformers>=4.36,<4.46",
+    "transformers>=4.36.0,<4.46.0",
     "optimum~=1.23",
     "datasets>=1.4.0",
     "sentencepiece",