depend on fixed/updated taming-transformers to fix import error

ZeroCool940711 · May 2, 2023 · 317fdc8 · 317fdc8
1 parent 39f843b
commit 317fdc8
Showing 1 changed file with 3 additions and 3 deletions.
diff --git 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/[email protected]",
         "transformers",
         "torchvision",
         "tqdm",