diff --git a/magvit2_pytorch/version.py b/magvit2_pytorch/version.py
index fc79d63..020ed73 100644
--- a/magvit2_pytorch/version.py
+++ b/magvit2_pytorch/version.py
@@ -1 +1 @@
-__version__ = '0.2.1'
+__version__ = '0.2.2'