Skip to content

fix: support RTX 5070 Ti (Blackwell/SM_100) + fix web UI voice upload bug - #8

Open
newlix wants to merge 1 commit into
neosun100:mainfrom
newlix:fix/rtx5070ti-blackwell-support
Open

fix: support RTX 5070 Ti (Blackwell/SM_100) + fix web UI voice upload bug#8
newlix wants to merge 1 commit into
neosun100:mainfrom
newlix:fix/rtx5070ti-blackwell-support

Conversation

@newlix

@newlix newlix commented Feb 20, 2026

Copy link
Copy Markdown

Summary

  • RTX 5070 Ti (Blackwell) support: upgrade CUDA 12.4 β†’ 12.8, PyTorch 2.3.1 β†’ 2.7.0, TensorRT β‰₯10.8.0, onnxruntime-gpu 1.18 β†’ 1.21
  • Fix ruamel.yaml incompatibility: pin ruamel.yaml==0.17.40 to fix AttributeError: 'Loader' object has no attribute 'max_depth' with HyperPyYAML 1.2.2
  • Fix web UI voice upload bug: frontend was sending prompt_audio but backend expected prompt_wav, causing all direct audio uploads to silently fail

Changes

Dockerfile & docker/Dockerfile

  • Base image: nvidia/cuda:12.4.1-cudnn-devel-ubuntu22.04 β†’ nvidia/cuda:12.8.1-cudnn-devel-ubuntu22.04

requirements.txt

  • PyTorch index: cu121 β†’ cu128
  • torch 2.3.1 β†’ 2.7.0
  • torchaudio 2.3.1 β†’ 2.7.0
  • onnxruntime-gpu 1.18.0 β†’ 1.21.0
  • tensorrt-cu12 pinned ==10.13.3.9 β†’ >=10.8.0 (supports Blackwell)
  • Add ruamel.yaml==0.17.40 (fixes startup crash with HyperPyYAML)
  • Remove legacy onnxruntime-cuda-12 private index (not needed for cu128)

app.py

  • Fix FormData field name: prompt_audio β†’ prompt_wav in web UI generate() function

Test plan

  • Build Docker image on CUDA 12.8 host with RTX 5070 Ti
  • Verify service starts without ruamel.yaml error
  • Verify web UI audio file upload works end-to-end
  • Verify voice cloning output quality unchanged

πŸ€– Generated with Claude Code

## Changes

### GPU Support (CUDA 12.8 + PyTorch 2.7.0)
- Dockerfile, docker/Dockerfile: bump base image nvidia/cuda:12.4.1 β†’ 12.8.1
- requirements.txt: cu121 β†’ cu128 index, torch 2.3.1 β†’ 2.7.0, torchaudio 2.3.1 β†’ 2.7.0
- requirements.txt: onnxruntime-gpu 1.18.0 β†’ 1.21.0 (CUDA 12.8 support)
- requirements.txt: tensorrt-cu12 pinned β†’ >=10.8.0 (Blackwell support)
- requirements.txt: pin ruamel.yaml==0.17.40 (fix HyperPyYAML incompatibility with 0.18+)
- requirements.txt: remove legacy onnxruntime-cuda-12 extra index (not needed for cu128)

### Bug Fix: Web UI voice upload
- app.py: fix FormData field name mismatch: prompt_audio β†’ prompt_wav
  (frontend was sending wrong field name, backend never received the uploaded file)

Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant