L0SG commited on
Commit
8fa3716
·
verified ·
1 Parent(s): 4164484

Use bundled ZeroGPU Mamba wheel

Browse files

Use the Hugging Face-provided mamba-ssm 2.3.2.post1 Blackwell wheel after validating the same package version locally with both Audex backends.

Files changed (2) hide show
  1. README.md +2 -0
  2. app.py +1 -1
README.md CHANGED
@@ -22,6 +22,8 @@ The demo includes **audio understanding, speech recognition, speech translation,
22
 
23
  Reasoning has no separate token cap by default for either model. `Max new tokens` sets a total cap shared by reasoning and the final answer.
24
 
 
 
25
  ## Run locally
26
 
27
  ```bash
 
22
 
23
  Reasoning has no separate token cap by default for either model. `Max new tokens` sets a total cap shared by reasoning and the final answer.
24
 
25
+ The hosted ZeroGPU runtime uses a prebuilt `mamba-ssm==2.3.2.post1` Blackwell wheel.
26
+
27
  ## Run locally
28
 
29
  ```bash
app.py CHANGED
@@ -55,7 +55,7 @@ WHEELS_DIR = Path(os.environ.get("AUDEX_WHEELS_DIR", str(DEFAULT_WHEELS_DIR)))
55
  BUILD_DIR = Path(os.environ.get("AUDEX_BUILD_DIR", str(DEFAULT_BUILD_DIR)))
56
  REPO_ID = os.environ.get("SPACE_ID", "L0SG/Nemotron-Labs-Audex")
57
  CAUSAL_CONV1D_VERSION = "1.6.2.post1"
58
- MAMBA_SSM_VERSION = "2.3.1"
59
  CAUSAL_CONV1D_SPEC = f"causal-conv1d=={CAUSAL_CONV1D_VERSION}"
60
  MAMBA_SSM_SPEC = f"mamba-ssm=={MAMBA_SSM_VERSION}"
61
  TORCH_ARCH = os.environ.get("AUDEX_TORCH_ARCH", "12.0" if IS_ZERO_GPU else "")
 
55
  BUILD_DIR = Path(os.environ.get("AUDEX_BUILD_DIR", str(DEFAULT_BUILD_DIR)))
56
  REPO_ID = os.environ.get("SPACE_ID", "L0SG/Nemotron-Labs-Audex")
57
  CAUSAL_CONV1D_VERSION = "1.6.2.post1"
58
+ MAMBA_SSM_VERSION = "2.3.2.post1"
59
  CAUSAL_CONV1D_SPEC = f"causal-conv1d=={CAUSAL_CONV1D_VERSION}"
60
  MAMBA_SSM_SPEC = f"mamba-ssm=={MAMBA_SSM_VERSION}"
61
  TORCH_ARCH = os.environ.get("AUDEX_TORCH_ARCH", "12.0" if IS_ZERO_GPU else "")