From 317fdc80480394e8511dc8f36930b043043c4d19 Mon Sep 17 00:00:00 2001
From: Andrew Powers-Holmes <aholmes@omnom.net>
Date: Tue, 2 May 2023 06:20:12 +0000
Subject: [PATCH] depend on fixed/updated taming-transformers to fix import
 error

---
 setup.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

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",