diff --git a/scripts/download_wheels.sh b/scripts/download_wheels.sh index d1a9f7b..e736a44 100644 --- a/scripts/download_wheels.sh +++ b/scripts/download_wheels.sh @@ -1,7 +1,7 @@ #!/bin/bash # Set variables -AWQ_KERNELS_VERSION="0.0.3" +AWQ_KERNELS_VERSION="0.0.5" RELEASE_URL="https://api.github.com/repos/casper-hansen/AutoAWQ_kernels/releases/tags/v${AWQ_KERNELS_VERSION}" # Create a directory to download the wheels diff --git a/setup.py b/setup.py index b8b31d6..e76586d 100644 --- a/setup.py +++ b/setup.py @@ -7,7 +7,7 @@ os.environ["CC"] = "g++" os.environ["CXX"] = "g++" -AUTOAWQ_KERNELS_VERSION = "0.0.4" +AUTOAWQ_KERNELS_VERSION = "0.0.5" PYPI_BUILD = os.getenv("PYPI_BUILD", "0") == "1" CUDA_VERSION = os.getenv("CUDA_VERSION", None) or torch.version.cuda ROCM_VERSION = os.environ.get("ROCM_VERSION", None) or torch.version.hip