diff --git a/dalle2_pytorch/version.py b/dalle2_pytorch/version.py
index 1f356cc..cd7ca49 100644
--- a/dalle2_pytorch/version.py
+++ b/dalle2_pytorch/version.py
@@ -1 +1 @@
-__version__ = '1.0.0'
+__version__ = '1.0.1'