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