Skip to content

Commit

Permalink
...
Browse files Browse the repository at this point in the history
  • Loading branch information
turboderp committed Jan 22, 2024
1 parent c872d5a commit 654912a
Showing 1 changed file with 8 additions and 6 deletions.
14 changes: 8 additions & 6 deletions .github/workflows/build-wheels-release.yml
Original file line number Diff line number Diff line change
Expand Up @@ -72,9 +72,11 @@ jobs:
if (!(mamba list cuda)[-1].contains('cuda')) {throw 'CUDA Toolkit failed to install!'}
if ([version]$env:CUDAVER -lt [version]'11.8.0') {$torchver = "torch==2.0.1"} else {$torchver = "torch==2.1.0"}
python -m pip install $torchver --index-url https://download.pytorch.org/whl/cu$cudaVersionPytorch
python3 -m pip install $torchver --index-url https://download.pytorch.org/whl/cu$cudaVersionPytorch
python -m pip install build wheel safetensors sentencepiece ninja
# python -m pip install build wheel safetensors sentencepiece ninja
pip3 install --upgrade build wheel safetensors sentencepiece ninja


- name: Build Wheel
id: build-wheel
Expand All @@ -98,7 +100,7 @@ jobs:
$BUILDTAG = "+cu$cudaVersionPytorch"
$env:TORCH_CUDA_ARCH_LIST = if ([version]$env:CUDAVER -lt [version]'11.8') {'6.0 6.1 7.0 7.5 8.0 8.6+PTX'} else {'6.0 6.1 7.0 7.5 8.0 8.6 8.9 9.0+PTX'}
python -m build -n --wheel -C--build-option=egg_info "-C--build-option=--tag-build=$BUILDTAG"
python3 -m build -n --wheel -C--build-option=egg_info "-C--build-option=--tag-build=$BUILDTAG"
- uses: actions/upload-artifact@v3
with:
Expand Down Expand Up @@ -159,8 +161,8 @@ jobs:

- name: Install Dependencies
run: |
python -m pip install torch==2.0.1 --index-url https://download.pytorch.org/whl/cpu
python -m pip install build wheel ninja
python3 -m pip install torch==2.0.1 --index-url https://download.pytorch.org/whl/cpu
python3 -m pip install build wheel ninja
- name: Build Wheel
id: build-wheel-sdist
Expand All @@ -176,7 +178,7 @@ jobs:
}
$env:EXLLAMA_NOCOMPILE=1
python -m build -n
python3 -m build -n
- uses: actions/upload-artifact@v3
with:
Expand Down

0 comments on commit 654912a

Please sign in to comment.