diff --git a/requirements.txt b/requirements.txt
index ccb8262..546a397 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,4 +1,4 @@
-torch==2.1.2
+torch==2.2.0
 pytorch_lightning==1.9.0
 transformers==4.36.0
 umap-learn==0.5.3
diff --git a/setup.py b/setup.py
index d9cfa5c..1cddc91 100644
--- a/setup.py
+++ b/setup.py
@@ -42,7 +42,7 @@
         "matplotlib",
         "panel",
         "einops",
-        "torch==2.1.2",
+        "torch==2.2.0",
         "datasets==2.16.1",
         "bitsandbytes==0.42.0",
         "xformers==0.0.23.post1",