run in docker

Signed-off-by: jiqing-feng <jiqing.feng@intel.com>

Author: jiqing-feng

.github/workflows/test_ipex.yml

@@ -18,38 +18,24 @@ jobs:
     strategy:
       fail-fast: false
       matrix:
-        transformers-version: ["4.47.0", "4.47.1"]
-        torch-version: ["2.4.0", "2.5.*"]
+        torch-version: ["2.6.0"]
+        transformers-version: ["4.47.*"]
 
-    runs-on: ubuntu-22.04
+    runs-on: ubuntu-latest
+    container: intel/intel-extension-for-pytorch:2.6.0-pip-base
 
     steps:
-      - name: Checkout code
-        uses: actions/checkout@v4
-
-      - name: Setup Python
-        uses: actions/setup-python@v5
-        with:
-          python-version: "3.10"
-
       - name: Install dependencies
         run: |
           pip install --upgrade pip
-          pip install torch==${{ matrix.torch-version }} torchaudio torchvision --extra-index-url https://download.pytorch.org/whl/cpu
-          pip install .[ipex,tests] transformers[testing]==${{ matrix.transformers-version }} intel_extension_for_pytorch==${{ matrix.torch-version }}
+          pip install .[ipex,tests] transformers[testing]==${{ matrix.transformers-version }}
 
       - name: Install bitsandbytes
         run: |
           git clone --branch multi-backend-refactor https://github.com/bitsandbytes-foundation/bitsandbytes.git
           cd bitsandbytes
           pip install .
 
-      - name: Install autoawq
-        run: |
-          git clone --branch setup https://github.com/jiqing-feng/AutoAWQ.git
-          cd AutoAWQ
-          pip install .
-
       - name: Assert versions
         run: |
           python -c "import torch; print(torch.__version__); assert torch.__version__.startswith('${{ matrix.torch-version }}'.replace('.*', ''))"