Skip to content

Commit

Permalink
try again
Browse files Browse the repository at this point in the history
  • Loading branch information
turboderp committed Jan 22, 2024
1 parent 33e90ca commit c872d5a
Show file tree
Hide file tree
Showing 2 changed files with 3 additions and 5 deletions.
2 changes: 1 addition & 1 deletion .github/workflows/build-wheels-release-rocm.yml
Original file line number Diff line number Diff line change
Expand Up @@ -63,7 +63,7 @@ jobs:
run: |
pip3 install torch==2.1.0 --index-url="https://download.pytorch.org/whl/rocm$ROCM_VERSION"
pip3 install --upgrade build wheel safetensors sentencepiece ninja
- name: Build Wheel
id: build-wheel
run: |
Expand Down
6 changes: 2 additions & 4 deletions .github/workflows/build-wheels-release.yml
Original file line number Diff line number Diff line change
Expand Up @@ -74,9 +74,7 @@ jobs:
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
# python -m pip install build wheel safetensors sentencepiece ninja
pip3 install --upgrade build wheel safetensors sentencepiece ninja

python -m pip install build wheel safetensors sentencepiece ninja
- name: Build Wheel
id: build-wheel
Expand All @@ -100,7 +98,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'}
python3 -m build -n --wheel -C--build-option=egg_info "-C--build-option=--tag-build=$BUILDTAG"
python -m build -n --wheel -C--build-option=egg_info "-C--build-option=--tag-build=$BUILDTAG"
- uses: actions/upload-artifact@v3
with:
Expand Down

0 comments on commit c872d5a

Please sign in to comment.