[PR run-slow] (huggingface#33939)

* force latest torch * Update .github/workflows/self-pr-slow-ci.yml Co-authored-by: Yih-Dar <[email protected]> --------- Co-authored-by: Yih-Dar <[email protected]>
fwtan · Oct 4, 2024 · 2e719e3 · 2e719e3
1 parent 061c2c4
commit 2e719e3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/.github/workflows/self-pr-slow-ci.yml b/.github/workflows/self-pr-slow-ci.yml
@@ -94,7 +94,7 @@ jobs:
 
       - name: Reinstall transformers in edit mode (remove the one installed during docker image build)
         working-directory: /transformers
-        run: python3 -m pip uninstall -y transformers && python3 -m pip install -e .
+        run: python3 -m pip uninstall -y transformers && python3 -m pip install -e . && python3 -m pip install --upgrade torch torchaudio torchvision
 
       - name: NVIDIA-SMI
         run: |