Spaces:
Running on Zero
Running on Zero
Use bundled ZeroGPU Mamba wheel
Browse filesUse the Hugging Face-provided mamba-ssm 2.3.2.post1 Blackwell wheel after validating the same package version locally with both Audex backends.
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.
|
| 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 "")
|