diff --git a/setup.py b/setup.py
index 9774ae2..815f7f7 100644
--- a/setup.py
+++ b/setup.py
@@ -27,10 +27,10 @@
         "omegaconf>=2.3.0",
         "pillow",
         "sentencepiece",
-        "torch>=1.6",
+        "torch>=2.0",
         "torchmetrics<0.8.0",
-        "pytorch-lightning<=1.7.7",
-        "taming-transformers>=0.0.1",
+        "pytorch-lightning>=2.0.0",
+        "taming-transformers @ git+https://github.com/neggles/taming-transformers.git@v0.0.2",
         "transformers",
         "torchvision",
         "tqdm",