diff --git a/setup.py b/setup.py index 92e81e3..9774ae2 100644 --- a/setup.py +++ b/setup.py @@ -32,7 +32,6 @@ "pytorch-lightning<=1.7.7", "taming-transformers>=0.0.1", "transformers", - "torch>=1.6", "torchvision", "tqdm", "vector-quantize-pytorch>=0.10.14",