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 && ( +
+ + {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 && (