update setup

huggingface · Oct 24, 2024 · 5f00574 · 5f00574
1 parent 8a77bd7
commit 5f00574
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 3 deletions.
diff --git a/.github/workflows/test_openvino.yml b/.github/workflows/test_openvino.yml
@@ -37,6 +37,7 @@ jobs:
           python -m pip install --upgrade pip
           # install PyTorch CPU version to avoid installing CUDA packages on GitHub runner without GPU
           pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu
+          pip install transformers==${{ matrix.transformers-version }}
           pip install .[openvino,openvino-tokenizers,tests,diffusers] onnxruntime
 
       - if: ${{ matrix.transformers-version == '4.36.0' }}

diff --git a/.github/workflows/test_openvino_basic.yml b/.github/workflows/test_openvino_basic.yml
@@ -61,6 +61,7 @@ jobs:
           # Install openvino manually to prevent dependency conflicts when .[openvino] pins
           # optimum or transformers to a specific version
           pip install ${{ matrix.openvino }}
+          pip install transformers==${{ matrix.transformers-version }}
           pip install .[tests]
 
       - name: Pip freeze

diff --git a/setup.py b/setup.py
@@ -28,8 +28,7 @@
 
 INSTALL_REQUIRE = [
     "torch>=1.11",
-    "transformers @ git+https://github.com/huggingface/transformers.git",
-    #"transformers>=4.36,<4.46",
+    "transformers>=4.36,<4.47",
     "optimum~=1.23",
     "datasets>=1.4.0",
     "sentencepiece",
@@ -61,7 +60,7 @@
 QUALITY_REQUIRE = ["black~=23.1", "ruff==0.4.4"]
 
 EXTRAS_REQUIRE = {
-    "neural-compressor": ["neural-compressor[pt]>3.0", "accelerate"],
+    "neural-compressor": ["neural-compressor[pt]>3.0", "accelerate", "transformers<4.46"],
     "openvino": [
         "openvino==2024.4.1.dev20240926",
         "nncf>=2.11.0",