diff --git a/.github/workflows/build-wheels.yml b/.github/workflows/build-wheels.yml
index 8e44060d..25c13b70 100644
--- 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
index 415147fa..d03924ad 100644
--- a/setup.py
+++ b/setup.py
@@ -12,7 +12,7 @@
 
 setup(
     name="exllama",
-    version="0.0.7",
+    version="0.0.8",
     install_requires=[
         "torch",
     ],