fix trainer

huggingface · Nov 3, 2023 · a328aa4 · a328aa4
1 parent 0ad7dc2
commit a328aa4
Show file tree

Hide file tree

Showing 3 changed files with 398 additions and 198 deletions.
diff --git a/.github/workflows/test_inc.yml b/.github/workflows/test_inc.yml
@@ -30,7 +30,8 @@ jobs:
     - name: Install dependencies
       run: |
         python -m pip install --upgrade pip
-        pip install .[neural-compressor,ipex,diffusers,tests]
+        pip install .[neural-compressor,diffusers,tests]
+        pip install intel-extension-for-pytorch
     - name: Test with Pytest
       run: |
         pytest tests/neural_compressor/