diff --git a/setup.py b/setup.py index 835850c..7915abe 100644 --- a/setup.py +++ b/setup.py @@ -28,7 +28,7 @@ "pillow", "sentencepiece", "torch>=1.6", - "taming-transformers>=0.0.1" + "taming-transformers>=0.0.1", "transformers", "torch>=1.6", "torchvision",