turboderp-org/exllamav2
1
0
Fork 0
mirror of https://github.com/turboderp-org/exllamav2.git synced 2026-04-20 06:19:00 +00:00
Code Issues Packages Projects Releases Wiki Activity

Use Torch 2.0.1 for cu117

Browse Source
This commit is contained in:
turboderp
2024-02-02 19:02:01 +01:00
parent f40e8374e9
commit efacbc53ca
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
.github/workflows/build-wheels-release.yml vendored
View File

@@ -1,4 +1,4 @@
name: Build Wheels & Release
pip name: Build Wheels & Release
on:
workflow_dispatch:
@@ -71,7 +71,7 @@ jobs:
if (!(mamba list cuda)[-1].contains('cuda')) {sleep -s 10; mamba install -y 'cuda' $cudaChannels.TrimEnd().Split()}
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.2.0"} else {$torchver = "torch==2.2.0"}
if ([version]$env:CUDAVER -lt [version]'11.8.0') {$torchver = "torch==2.0.1"} else {$torchver = "torch==2.2.0"}
python -m pip install $torchver --index-url https://download.pytorch.org/whl/cu$cudaVersionPytorch
python -m pip install --upgrade setuptools==65.5.1