test

huggingface · Oct 7, 2024 · 5013c2e · 5013c2e
1 parent 6ff507f
commit 5013c2e
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/.github/workflows/test_export_onnx_cli.yml b/.github/workflows/test_export_onnx_cli.yml
@@ -32,6 +32,7 @@ jobs:
         run: |
           pip install --no-cache-dir torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu
           pip install .[tests,exporters]
+          pip install git+https://github.com/huggingface/accelerate.git
 
       - name: Test with pytest (in parallel)
         working-directory: tests