peft in testing

huggingface · Oct 28, 2024 · dcb140d · dcb140d
1 parent 96dedd8
commit dcb140d
Show file tree

Hide file tree

Showing 4 changed files with 3 additions and 4 deletions.
diff --git a/.github/workflows/test_ipex.yml b/.github/workflows/test_ipex.yml
@@ -20,7 +20,7 @@ jobs:
       matrix:
         torch-version: ["2.2.0", "2.3.*", "2.4.*"]
         transformers-version: ["4.39.0", "4.44.*"]
-        python-version: [3.9]
+        python-version: ["3.9"]
         os: [ubuntu-latest]
 
     runs-on: ${{ matrix.os }}

diff --git a/.github/workflows/test_openvino.yml b/.github/workflows/test_openvino.yml
@@ -46,7 +46,7 @@ jobs:
         run: |
           pip install --upgrade pip
           pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu
-          pip install .[openvino,openvino-tokenizers,diffusers,peft,tests] transformers[testing]==${{ matrix.transformers-version }}
+          pip install .[openvino,openvino-tokenizers,diffusers,tests] transformers[testing]==${{ matrix.transformers-version }}
 
       - name: Assert versions
         run: |

diff --git a/.github/workflows/test_openvino_slow.yml b/.github/workflows/test_openvino_slow.yml
@@ -24,7 +24,6 @@ jobs:
     strategy:
       fail-fast: false
       matrix:
-        # TODO: testong python upper bound seems unnecessary
         # Testing lower and upper bound of supported Python versions
         # This also ensures that the test fails if dependencies break for Python 3.7
         python-version: ["3.9", "3.12"]

diff --git a/setup.py b/setup.py
@@ -55,6 +55,7 @@
     "tiktoken",
     "sentence-transformers",
     "open_clip_torch>=2.26.1",
+    "peft",
 ]
 
 QUALITY_REQUIRE = ["black~=23.1", "ruff==0.4.4"]
@@ -71,7 +72,6 @@
     "diffusers": ["diffusers"],
     "quality": QUALITY_REQUIRE,
     "tests": TESTS_REQUIRE,
-    "peft": ["peft"],
 }
 
 setup(