use cpu image

huggingface · Dec 9, 2024 · 214c69f · 214c69f
1 parent 9087bc7
commit 214c69f
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 13 deletions.
diff --git a/.github/workflows/test_cli_cpu_ipex.yaml b/.github/workflows/test_cli_cpu_ipex.yaml
@@ -33,19 +33,15 @@ jobs:
     runs-on:
       group: aws-c7i-8xlarge-plus
 
+    container:
+      image: ghcr.io/huggingface/optimum-benchmark:latest-cpu
+
     steps:
       - name: Checkout
         uses: actions/checkout@v4
 
-      - name: Set up Python 3.10
-        uses: actions/setup-python@v5
-        with:
-          python-version: "3.10"
-
       - name: Install requirements
         run: |
-          pip install --upgrade pip
-          pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu
           pip install -e .[testing,ipex,diffusers,timm]
 
       - name: Run tests

diff --git a/.github/workflows/test_cli_cpu_openvino.yaml b/.github/workflows/test_cli_cpu_openvino.yaml
@@ -33,18 +33,15 @@ jobs:
     runs-on:
       group: aws-c7i-8xlarge-plus
 
+    container:
+      image: ghcr.io/huggingface/optimum-benchmark:latest-cpu
+
     steps:
       - name: Checkout
         uses: actions/checkout@v4
 
-      - name: Set up Python 3.10
-        uses: actions/setup-python@v5
-        with:
-          python-version: "3.10"
-
       - name: Install requirements
         run: |
-          pip install --upgrade pip
           pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu
           pip install -e .[testing,openvino,diffusers,timm]