Switch to Pytorch 2.1

0cc4m · Nov 12, 2023 · 5b028ce · 5b028ce
1 parent 1fd706b
commit 5b028ce
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/.github/workflows/build-wheels.yml b/.github/workflows/build-wheels.yml
@@ -33,7 +33,7 @@ jobs:
 
       - name: Install Dependencies
         run: |
-          mamba install -y 'pytorch[version=2.0.0,build=py*_cuda11.8*]' 'pytorch-cuda=11.8' 'sentencepiece' 'cuda' 'ninja' -c 'pytorch' -c 'nvidia/label/cuda-11.8.0' -c 'nvidia' -c 'conda-forge' -c 'defaults'
+          mamba install -y 'pytorch[version=2.1.0,build=py*_cuda11.8*]' 'pytorch-cuda=11.8' 'sentencepiece' 'cuda' 'ninja' -c 'pytorch' -c 'nvidia/label/cuda-11.8.0' -c 'nvidia' -c 'conda-forge' -c 'defaults'
           python -m pip install build transformers==4.28.0 datasets safetensors accelerate==0.18.0
 
       - name: Build Wheel

diff --git a/setup.py b/setup.py
@@ -12,7 +12,7 @@
 
 setup(
     name="exllama",
-    version="0.0.7",
+    version="0.0.8",
     install_requires=[
         "torch",
     ],