diff --git a/awq/__init__.py b/awq/__init__.py
index 2160a0f4..1870e2bd 100644
--- a/awq/__init__.py
+++ b/awq/__init__.py
@@ -1,2 +1,2 @@
-__version__ = "0.2.3"
+__version__ = "0.2.4"
 from awq.models.auto import AutoAWQForCausalLM
diff --git a/scripts/download_wheels.sh b/scripts/download_wheels.sh
index 70939155..9f2c43a3 100644
--- a/scripts/download_wheels.sh
+++ b/scripts/download_wheels.sh
@@ -1,7 +1,7 @@
 #!/bin/bash
 
 # Set variables
-AWQ_VERSION="0.2.3"
+AWQ_VERSION="0.2.4"
 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 a46b9443..c02ff895 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.3"
+AUTOAWQ_VERSION = "0.2.4"
 PYPI_BUILD = os.getenv("PYPI_BUILD", "0") == "1"
 
 CUDA_VERSION = os.getenv("CUDA_VERSION", None) or torch.version.cuda