From 145fc22ed3b98ec2a2c61c4ab613ad6327a4548e Mon Sep 17 00:00:00 2001 From: Song Yanzeng Date: Thu, 2 Apr 2026 15:33:52 +0800 Subject: [PATCH] Use environment variable for BASE_WHEEL_URL Allow overriding the wheel download host (e.g., for mirrors or proxies) --- setup.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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