diff --git a/awq/__init__.py b/awq/__init__.py
index b8a4b6a4..55e3af11 100644
--- a/awq/__init__.py
+++ b/awq/__init__.py
@@ -1,2 +1,2 @@
-__version__ = "0.2.0"
+__version__ = "0.2.1"
 from awq.models.auto import AutoAWQForCausalLM
diff --git a/scripts/download_wheels.sh b/scripts/download_wheels.sh
index b5b56e63..1d5e569d 100644
--- a/scripts/download_wheels.sh
+++ b/scripts/download_wheels.sh
@@ -1,7 +1,7 @@
 #!/bin/bash
 
 # Set variables
-AWQ_VERSION="0.2.0"
+AWQ_VERSION="0.2.1"
 RELEASE_URL="https://api.github.com/repos/casper-hansen/AutoAWQ/releases/tags/v${AWQ_VERSION}"
 
 # Create a directory to download the wheels
diff --git a/setup.py b/setup.py
index 63641124..45111f77 100644
--- a/setup.py
+++ b/setup.py
@@ -31,7 +31,7 @@ def get_kernels_whl_url(
     return f"https://github.com/casper-hansen/AutoAWQ_kernels/releases/download/v{release_version}/autoawq_kernels-{release_version}+{gpu_system_version}-cp{python_version}-cp{python_version}-{platform}_{architecture}.whl"
 
 
-AUTOAWQ_VERSION = "0.2.0"
+AUTOAWQ_VERSION = "0.2.1"
 PYPI_BUILD = os.getenv("PYPI_BUILD", "0") == "1"
 
 CUDA_VERSION = os.getenv("CUDA_VERSION", None) or torch.version.cuda