diff --git a/.bumpversion.cfg b/.bumpversion.cfg
index 73ee5be..add0569 100644
--- a/.bumpversion.cfg
+++ b/.bumpversion.cfg
@@ -1,5 +1,5 @@
 [bumpversion]
-current_version = 0.1.6
+current_version = 0.1.7
 commit = True
 tag = True
 parse = (?P<major>\d+)\.(?P<minor>\d+)\.(?P<patch>\d+)
diff --git a/prototorch/models/__init__.py b/prototorch/models/__init__.py
index c6c1313..37a4b5b 100644
--- a/prototorch/models/__init__.py
+++ b/prototorch/models/__init__.py
@@ -7,4 +7,4 @@ from .knn import KNN
 from .neural_gas import NeuralGas
 from .vis import *
 
-__version__ = "0.1.6"
+__version__ = "0.1.7"
diff --git a/setup.py b/setup.py
index d2d045e..ce64657 100644
--- a/setup.py
+++ b/setup.py
@@ -27,7 +27,7 @@ ALL = DEV + EXAMPLES + TESTS
 
 setup(
     name=safe_name("prototorch_" + PLUGIN_NAME),
-    version="0.1.6",
+    version="0.1.7",
     description="Pre-packaged prototype-based "
     "machine learning models using ProtoTorch and PyTorch-Lightning.",
     long_description=long_description,