diff --git a/setup.py b/setup.py index 80d57ef7f..5d81e7376 100755 --- a/setup.py +++ b/setup.py @@ -34,7 +34,7 @@ PACKAGE_NAME = "mamba_ssm" -BASE_WHEEL_URL = "https://github.com/state-spaces/mamba/releases/download/{tag_name}/{wheel_name}" +BASE_WHEEL_URL = os.getenv("MAMBA_BASE_WHEEL_URL", "https://github.com/state-spaces/mamba/releases/download/{tag_name}/{wheel_name}") # FORCE_BUILD: Force a fresh build locally, instead of attempting to find prebuilt wheels # SKIP_CUDA_BUILD: Intended to allow CI to use a simple `python setup.py sdist` run to copy over raw files, without any cuda compilation