Update build_wheels_release_python312test.yml

turboderp · Jun 6, 2024 · dad511a · dad511a
1 parent 5b5f162
commit dad511a
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/.github/workflows/build_wheels_release_python312test.yml b/.github/workflows/build_wheels_release_python312test.yml
@@ -143,6 +143,7 @@ jobs:
 
           # --- Install dependencies
 
+          python3 -m ensurepip --upgrade
           pip3 install torch==${{ matrix.torch }} --index-url="https://download.pytorch.org/whl/rocm$ROCM_VERSION"
           pip3 install --upgrade build wheel safetensors sentencepiece ninja
           pip3 install --upgrade setuptools==65.5.0
@@ -188,6 +189,7 @@ jobs:
           
           # --- Install dependencies
           
+          python -m ensurepip --upgrade
           python -m pip install torch==${{ matrix.torch }} --index-url https://download.pytorch.org/whl/cu$cudaVersionPytorch
           python -m pip install --upgrade setuptools==65.5.1
           python -m pip install --upgrade build wheel safetensors sentencepiece ninja