Dispatch

.github/workflows/test.yml

@@ -9,6 +9,7 @@ on:
       - "examples/**"
       - "setup.py"
     types: [opened, synchronize, reopened]
+  workflow_dispatch:
 
 env:
   HF_HOME: ~/hf_cache
@@ -18,6 +19,8 @@ env:
 jobs:
   run-tests:
     runs-on: ubuntu-latest
+    container:
+      image: huggingface/accelerate:cpu-nightly
     strategy:
       fail-fast: false
       matrix:
@@ -38,11 +41,6 @@ jobs:
           test_rest
         ]
     steps:
-    - uses: actions/checkout@v3.1.0
-    - name: Set up python 3.8
-      uses: actions/setup-python@v3
-      with:
-        python-version: 3.8
     
     - name: Install the library
       run: |

docker/accelerate-cpu/Dockerfile

@@ -17,9 +17,11 @@ RUN python3 -m venv ${VIRTUAL_ENV}
 # Make sure we use the virtualenv
 ENV PATH="${VIRTUAL_ENV}/bin:$PATH"
 WORKDIR /workspace
+# Setup UV
+RUN pip install uv --no-cache-dir
 # Install specific CPU torch wheel to save on space
-RUN python3 -m pip install --upgrade --no-cache-dir pip
-RUN python3 -m pip install --no-cache-dir \
+RUN uv pip install --upgrade --no-cache-dir pip
+RUN uv pip install --no-cache-dir \
     jupyter \
     git+https://github.com/huggingface/accelerate#egg=accelerate[testing,test_trackers] \
     --extra-index-url https://download.pytorch.org/whl/cpu