Actions: Update install methods (#775)

Signed-off-by: kingbri <8082010+kingbri1@users.noreply.github.com>

Actions: Test

Author: kingbri1

.github/workflows/build-wheels-release-rocm62.yml

@@ -60,14 +60,8 @@ jobs:
           large-packages: false
           swap-storage: true
 
-      # Setup Python
-
       - uses: actions/checkout@v4
 
-      - uses: actions/setup-python@v5.4.0
-        with:
-          python-version: ${{ matrix.pyver }}
-
       # Get version string from package
 
       - name: Get version string
@@ -85,11 +79,11 @@ jobs:
             Write-Output "PACKAGE_VERSION=None" >> "$env:GITHUB_OUTPUT"
           }
 
-      # Pin VS build tools to 17.9 so builds won't fail
-
-      - name: Install VS2022 BuildTools 17.9.7
-        run: choco install -y visualstudio2022buildtools --version=117.9.7.0 --params "--add Microsoft.VisualStudio.Component.VC.Tools.x86.x64 --installChannelUri https://aka.ms/vs/17/release/180911598_-255012421/channel"
-        if: runner.os == 'Windows'
+      # Install uv for easier python setup
+      - name: Install the latest version of uv and set the python version
+        uses: astral-sh/setup-uv@v5
+        with:
+          python-version: ${{ matrix.pyver }}
 
       # Install ROCm SDK, apparently needs to happen before setting up Python
 
@@ -118,101 +112,15 @@ 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 setuptools==69.5.1 build wheel safetensors sentencepiece ninja
-          pip3 cache purge
+          uv pip install torch==${{ matrix.torch }} --index-url="https://download.pytorch.org/whl/rocm$ROCM_VERSION"
+          uv pip install --upgrade build setuptools==69.5.1 wheel packaging ninja safetensors sentencepiece tokenizers numpy
 
           # --- Build wheel 
 
           python3 -m build -n --wheel -C--build-option=egg_info "-C--build-option=--tag-build=+rocm${{ matrix.rocm }}-torch${{ matrix.torch }}"
 
-      # Build for CUDA
-
-      - name: Setup Mamba
-        if: matrix.cuda != ''
-        uses: conda-incubator/setup-miniconda@v3.1.0
-        with:
-          activate-environment: "exllama"
-          python-version: ${{ matrix.pyver }}
-#          miniforge-variant: Mambaforge
-          miniforge-version: latest
-#          use-mamba: true
-          add-pip-as-python-dependency: true
-          auto-activate-base: false
-
-      - name: Build for CUDA
-        if: matrix.cuda != ''
-        run: |
-          # --- Spawn the VS shell
-          if ($IsWindows) {
-            Import-Module 'C:\Program Files (x86)\Microsoft Visual Studio\2022\BuildTools\Common7\Tools\Microsoft.VisualStudio.DevShell.dll'
-            Enter-VsDevShell -VsInstallPath 'C:\Program Files (x86)\Microsoft Visual Studio\2022\BuildTools' -DevCmdArguments '-arch=x64 -host_arch=x64'
-            $env:DISTUTILS_USE_SDK=1
-          }
-  
-          # --- Install CUDA using Conda
-          $cudaVersion = '${{ matrix.cuda }}'
-          $cudaVersionPytorch = '${{ matrix.cuda }}'.Remove('${{ matrix.cuda }}'.LastIndexOf('.')).Replace('.','')
-
-          $env:MAMBA_NO_LOW_SPEED_LIMIT = 1
-          mamba install -y -c nvidia/label/cuda-$cudaVersion cuda-toolkit cuda-runtime
-
-          if (!(mamba list cuda)[-1].contains('cuda')) {sleep -s 10; mamba install -y 'cuda' $cudaVersion}
-          if (!(mamba list cuda)[-1].contains('cuda')) {throw 'CUDA Toolkit failed to install!'}
-
-          $env:CUDA_PATH = $env:CONDA_PREFIX
-          $env:CUDA_HOME = $env:CONDA_PREFIX
-          if ($IsLinux) {$env:LD_LIBRARY_PATH = $env:CONDA_PREFIX + '/lib:' + $env:LD_LIBRARY_PATH}
-          
-          # --- 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==69.5.1 build wheel safetensors sentencepiece ninja
-
-          # --- Build wheel
-                  
-          $BUILDTAG = "+cu$cudaVersionPytorch-torch${{ matrix.torch }}"
-          $env:TORCH_CUDA_ARCH_LIST = '${{ matrix.cudaarch }}'
-          python -m build -n --wheel -C--build-option=egg_info "-C--build-option=--tag-build=$BUILDTAG"
-
-      # Build sdist
-
-      - name: Build sdist
-        if: matrix.cuda == '' && matrix.rocm == ''
-        run: |
-          # --- Spawn the VS shell
-          if ($IsWindows) {
-            Import-Module 'C:\Program Files (x86)\Microsoft Visual Studio\2022\BuildTools\Common7\Tools\Microsoft.VisualStudio.DevShell.dll'
-            Enter-VsDevShell -VsInstallPath 'C:\Program Files (x86)\Microsoft Visual Studio\2022\BuildTools' -DevCmdArguments '-arch=x64 -host_arch=x64'
-            $env:DISTUTILS_USE_SDK=1
-          }
-
-          # --- Install dependencies
-          
-          python -m pip install torch==${{ matrix.torch }} --index-url https://download.pytorch.org/whl/cpu
-          python -m pip install build wheel ninja
-
-          # --- Build wheel
-           
-          $env:EXLLAMA_NOCOMPILE=1
-          python -m build -n
-
       # Upload files
 
-      - uses: actions/upload-artifact@v4
-        if: matrix.artname == 'wheel'
-        with:
-          name: wheel-${{ matrix.os }}-py${{ matrix.pyver }}-cuda${{ matrix.cuda }}-torch${{ matrix.torch }}
-          path: ./dist/*
-
-      - uses: actions/upload-artifact@v4
-        if: matrix.artname == 'sdist'
-        with:
-          name: 'sdist'
-          path: ./dist/*
-
       - name: Upload files to GitHub release
         if: steps.package_version.outputs.PACKAGE_VERSION != 'None' && inputs.release == '1'
         uses: svenstaro/upload-release-action@2.6.1