fix accelerate version for test (#927)

* fix accelerate version for test * fix np for ipex
huggingface · Oct 8, 2024 · e0c6aaa · e0c6aaa
1 parent 542347b
commit e0c6aaa
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 0 deletions.
diff --git a/.github/workflows/test_ipex.yml b/.github/workflows/test_ipex.yml
@@ -43,6 +43,10 @@ jobs:
           pip install Pillow parameterized
           pip install transformers[testing]==${{ matrix.transformers-version }}
           pip install .[ipex]
+
+      - if: ${{ matrix.ipex-version == '2.2.0' }}
+        run: pip install numpy==1.*
+
       - name: Test with Pytest
         run: |
           pytest tests/ipex/
diff --git a/.github/workflows/test_openvino.yml b/.github/workflows/test_openvino.yml
@@ -40,6 +40,9 @@ jobs:
           pip install .[openvino,openvino-tokenizers,tests,diffusers] onnxruntime
           pip install transformers==${{ matrix.transformers-version }}
 
+      - if: ${{ matrix.transformers-version == '4.36.0' }}
+        run: pip install accelerate==0.*
+
       - name: Test with Pytest
         env:
           HF_HUB_READ_TOKEN: ${{ secrets.HF_HUB_READ_TOKEN }}