diff --git a/sgm/__init__.py b/sgm/__init__.py
index 24bc84a..2a05893 100644
--- a/sgm/__init__.py
+++ b/sgm/__init__.py
@@ -1,4 +1,4 @@
 from .models import AutoencodingEngine, DiffusionEngine
 from .util import get_configs_path, instantiate_from_config
 
-__version__ = "0.1.0"
+__version__ = "0.1.1"