@acpfx/tts-pocket 0.2.5 → 0.2.6

package/README.md

@@ -6,7 +6,7 @@ Local text-to-speech via Pocket TTS. A lightweight ~100M parameter model that ru
 
 This package is a pipeline node for [@acpfx/cli](../orchestrator/README.md). See the CLI package for installation and usage.
 
-The postinstall script downloads a prebuilt binary for your platform. Supported: macOS (Apple Silicon with Metal), Linux (x86_64, optional CUDA).
+The postinstall script downloads a prebuilt binary for your platform. On Linux/Windows with an NVIDIA GPU (Ampere or newer, compute capability 8.0+), a CUDA-accelerated binary is downloaded automatically. Otherwise falls back to CPU (~6x realtime on Apple Silicon).
 
 ## Manifest
 
@@ -31,12 +31,22 @@ nodes:
     outputs: [player]
 ```
 
+## GPU Acceleration
+
+| Platform | Acceleration | Requirement |
+|----------|-------------|-------------|
+| macOS | Metal | Apple Silicon (automatic) |
+| Linux/Windows | CUDA | NVIDIA Ampere+ (RTX 3090, A100, etc.) |
+| All | CPU | Fallback, ~6x realtime on Apple Silicon |
+
 ## Building from Source
 
 ```bash
 cargo build --release -p node-tts-pocket
 # With Metal (macOS):
 cargo build --release -p node-tts-pocket --features metal
+# With CUDA (Linux/Windows, requires CUDA toolkit):
+cargo build --release -p node-tts-pocket --features cuda
 ```
 
 ## External Links

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@acpfx/tts-pocket",
-  "version": "0.2.5",
+  "version": "0.2.6",
   "description": "Local TTS via Pocket TTS (on-device, Rust/Candle, CPU-capable)",
   "repository": {
     "type": "git",

package/scripts/postinstall.js

@@ -13,12 +13,15 @@ const SUPPORTED_PLATFORMS = {
   "win32 x64": "tts-pocket-win32-x64.exe",
 };
 
-function hasNvidiaGpu() {
+function getCudaComputeCap() {
   try {
-    require("child_process").execSync("nvidia-smi", { stdio: "ignore" });
-    return true;
+    const output = require("child_process")
+      .execSync("nvidia-smi --query-gpu=compute_cap --format=csv,noheader", { stdio: ["pipe", "pipe", "ignore"] })
+      .toString().trim();
+    // "8.6" -> 86, "7.5" -> 75
+    return parseInt(output.replace(".", ""), 10);
   } catch {
-    return false;
+    return 0;
   }
 }
 
@@ -32,8 +35,8 @@ function getBinaryName() {
     );
   }
 
-  // Try CUDA variant on Linux/Windows when an NVIDIA GPU is present
-  if ((process.platform === "linux" || process.platform === "win32") && hasNvidiaGpu()) {
+  // Try CUDA variant on Linux/Windows when an NVIDIA Ampere+ GPU is present (compute cap >= 8.0)
+  if ((process.platform === "linux" || process.platform === "win32") && getCudaComputeCap() >= 80) {
     return base.replace(/(\.\w+)?$/, "-cuda$1");
   }
   return base;