diff --git a/.github/workflows/release.yml b/.github/workflows/release.yml
index 1312ca37..a3c224df 100644
--- a/.github/workflows/release.yml
+++ b/.github/workflows/release.yml
@@ -191,10 +191,10 @@ jobs:
pip install --no-deps chatterbox-tts
pip install --no-deps hume-tada
- - name: Install PyTorch with CUDA 12.6
+ - name: Install PyTorch with CUDA 12.8
run: |
- pip install torch --index-url https://download.pytorch.org/whl/cu126 --force-reinstall --no-deps
- pip install torchaudio --index-url https://download.pytorch.org/whl/cu126 --force-reinstall --no-deps
+ pip install torch --index-url https://download.pytorch.org/whl/cu128 --force-reinstall --no-deps
+ pip install torchaudio --index-url https://download.pytorch.org/whl/cu128 --force-reinstall --no-deps
- name: Verify CUDA support in torch
run: |
@@ -211,8 +211,8 @@ jobs:
python scripts/package_cuda.py \
backend/dist/voicebox-server-cuda/ \
--output release-assets/ \
- --cuda-libs-version cu126-v1 \
- --torch-compat ">=2.6.0,<2.11.0"
+ --cuda-libs-version cu128-v1 \
+ --torch-compat ">=2.7.0,<2.11.0"
- name: Upload archives to GitHub Release
if: startsWith(github.ref, 'refs/tags/')
@@ -221,8 +221,8 @@ jobs:
files: |
release-assets/voicebox-server-cuda.tar.gz
release-assets/voicebox-server-cuda.tar.gz.sha256
- release-assets/cuda-libs-cu126-v1.tar.gz
- release-assets/cuda-libs-cu126-v1.tar.gz.sha256
+ release-assets/cuda-libs-cu128-v1.tar.gz
+ release-assets/cuda-libs-cu128-v1.tar.gz.sha256
release-assets/cuda-libs.json
draft: true
env:
diff --git a/app/src/components/ServerSettings/GpuAcceleration.tsx b/app/src/components/ServerSettings/GpuAcceleration.tsx
index b9cb70b3..7b2c9749 100644
--- a/app/src/components/ServerSettings/GpuAcceleration.tsx
+++ b/app/src/components/ServerSettings/GpuAcceleration.tsx
@@ -243,7 +243,40 @@ export function GpuAcceleration() {
{/* Native GPU detected - no CUDA download needed */}
- {/* CUDA download section - only show when no GPU is active (native or CUDA) */}
+ {/* Currently running CUDA - show switch back to CPU */}
+ {isCurrentlyCuda && platform.metadata.isTauri && (
+ <>
+ {restartPhase !== 'idle' ? (
+
+
+
+ {restartPhase === 'stopping' && 'Stopping server...'}
+ {restartPhase === 'waiting' && 'Restarting server...'}
+ {restartPhase === 'ready' && 'Server restarted successfully!'}
+
+
+ ) : (
+
+
+ Running with CUDA GPU acceleration. Switch back to CPU if needed (you can
+ re-download later).
+
+
+
+ )}
+ {error && (
+
+ )}
+ >
+ )}
+
+ {/* CUDA download/manage section - show when no native GPU and not currently running CUDA */}
{!hasNativeGpu && !isCurrentlyCuda && (
<>
{/* Download progress (manual download or auto-update) */}
@@ -315,7 +348,7 @@ export function GpuAcceleration() {
)}
{/* Downloaded but not active - show switch button */}
- {cudaAvailable && !isCurrentlyCuda && platform.metadata.isTauri && (
+ {cudaAvailable && platform.metadata.isTauri && (
CUDA backend is downloaded and ready. Restart the server to enable GPU
@@ -328,27 +361,8 @@ export function GpuAcceleration() {
)}
- {/* Currently active - show switch back to CPU */}
- {isCurrentlyCuda && platform.metadata.isTauri && (
-
-
- Running with CUDA GPU acceleration. Switch back to CPU if needed (you can
- re-download later).
-
-
-
- )}
-
{/* Delete option when downloaded (and not active) */}
- {cudaAvailable && !isCurrentlyCuda && (
+ {cudaAvailable && (