Unverified Commit ae146126 authored by Sayak Paul's avatar Sayak Paul Committed by GitHub
Browse files

[CI] uninstall deps properly from pr gpu tests. (#11102)

uninstall deps properly from pr gpu tests.
parent 0ab8fe49
...@@ -177,6 +177,7 @@ jobs: ...@@ -177,6 +177,7 @@ jobs:
torch_cuda_tests: torch_cuda_tests:
name: Torch CUDA Tests name: Torch CUDA Tests
needs: [check_code_quality, check_repository_consistency]
runs-on: runs-on:
group: aws-g4dn-2xlarge group: aws-g4dn-2xlarge
container: container:
...@@ -245,7 +246,7 @@ jobs: ...@@ -245,7 +246,7 @@ jobs:
run_examples_tests: run_examples_tests:
name: Examples PyTorch CUDA tests on Ubuntu name: Examples PyTorch CUDA tests on Ubuntu
pip uninstall transformers -y && python -m uv pip install -U transformers@git+https://github.com/huggingface/transformers.git --no-deps needs: [check_code_quality, check_repository_consistency]
runs-on: runs-on:
group: aws-g4dn-2xlarge group: aws-g4dn-2xlarge
...@@ -264,6 +265,7 @@ jobs: ...@@ -264,6 +265,7 @@ jobs:
- name: Install dependencies - name: Install dependencies
run: | run: |
python -m venv /opt/venv && export PATH="/opt/venv/bin:$PATH" python -m venv /opt/venv && export PATH="/opt/venv/bin:$PATH"
pip uninstall transformers -y && python -m uv pip install -U transformers@git+https://github.com/huggingface/transformers.git --no-deps
python -m uv pip install -e [quality,test,training] python -m uv pip install -e [quality,test,training]
- name: Environment - name: Environment
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment