diff --git a/dalle2_pytorch/version.py b/dalle2_pytorch/version.py
index 090480d..d2a3165 100644
--- a/dalle2_pytorch/version.py
+++ b/dalle2_pytorch/version.py
@@ -1 +1 @@
-__version__ = '0.15.3'
+__version__ = '0.15.4'