diff --git a/awq/__init__.py b/awq/__init__.py
index 446e7aad..950c6a41 100644
--- a/awq/__init__.py
+++ b/awq/__init__.py
@@ -1,2 +1,2 @@
-__version__ = "0.1.6"
+__version__ = "0.1.7"
 from awq.models.auto import AutoAWQForCausalLM
\ No newline at end of file
diff --git a/setup.py b/setup.py
index 3ae57dce..2fa9e28c 100644
--- a/setup.py
+++ b/setup.py
@@ -7,7 +7,7 @@
 
 os.environ["CC"] = "g++"
 os.environ["CXX"] = "g++"
-AUTOAWQ_VERSION = "0.1.6"
+AUTOAWQ_VERSION = "0.1.7"
 PYPI_BUILD = os.getenv("PYPI_BUILD", "0") == "1"
 
 if not PYPI_BUILD: