turboderp-org/exllamav2
1
0
Fork 0
mirror of https://github.com/turboderp-org/exllamav2.git synced 2026-03-15 00:07:26 +00:00
Code Issues Packages Projects Releases Wiki Activity

Bump Torch to 2.2.0

Browse Source
This commit is contained in:
turboderp
2024-02-02 18:43:25 +01:00
parent c0ddebaaaf
commit 3cd61203c5
3 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -61,7 +61,7 @@ jobs:
- name: Install Dependencies
run: |
pip3 install torch==2.1.0 --index-url="https://download.pytorch.org/whl/rocm$ROCM_VERSION"
pip3 install torch==2.2.0 --index-url="https://download.pytorch.org/whl/rocm$ROCM_VERSION"
pip3 install --upgrade build wheel safetensors sentencepiece ninja
- name: Build Wheel

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

@@ -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.0.1"} else {$torchver = "torch==2.1.0"}
if ([version]$env:CUDAVER -lt [version]'11.8.0') {$torchver = "torch==2.2.0"} 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
@@ -160,7 +160,7 @@ jobs:
- name: Install Dependencies
run: |
python -m pip install torch==2.1.0 --index-url https://download.pytorch.org/whl/cpu
python -m pip install torch==2.2.0 --index-url https://download.pytorch.org/whl/cpu
python -m pip install build wheel ninja
- name: Build Wheel

2
requirements.txt
View File

@@ -1,7 +1,7 @@
pandas
ninja
fastparquet
torch>=2.1.0
torch>=2.2.0
safetensors>=0.3.2
sentencepiece>=0.1.97
pygments