mirror of
https://github.com/JezzWTF/vibepod.git
synced 2026-06-01 15:22:14 +00:00
Claude
bb6da662de
Introduces a third hardware mode alongside CUDA and CPU: ROCm (AMD GPU). AMD GPUs present as CUDA devices under PyTorch ROCm, so the existing GPU path is reused with minimal changes — the main additions are wheel management, device detection, and suppressing flash_attn (unsupported on ROCm). - server/vibevoice_server.py: extend _resolve_device() to recognise 'rocm' (auto-detected via torch.version.hip); add _torch_device() helper that maps 'rocm' → 'cuda' for all PyTorch API calls; apply GPU optimisations for both cuda and rocm in _init_model(); always use sdpa on ROCm; propagate _torch_device() to _load_voice_presets() map_location. - server/start.sh: add --rocm flag; sync .venv-rocm with uv sync --no-sources then replace torch with the ROCm 6.2 wheel via uv pip install; set VIBEPOD_DEVICE=rocm for uvicorn. - server/pyproject.toml: register pytorch-rocm62 index (explicit); add .venv-rocm to ruff excludes. - package.json: add dev:rocm and dev:server:rocm scripts. - README.md: document ROCm mode, prerequisites (RX 6000+, ROCm 6.2+, Linux), and new commands; expand CUDA vs CPU section to CUDA vs CPU vs ROCm. https://claude.ai/code/session_0168pSswiaoEf6LEx6UQWfBu
24 lines
933 B
JSON
24 lines
933 B
JSON
{
|
|
"name": "vibepod",
|
|
"version": "0.1.0",
|
|
"private": true,
|
|
"scripts": {
|
|
"build": "pnpm --filter vibepod-web build",
|
|
"dev": "bash dev.sh",
|
|
"dev:cpu": "bash dev.sh --cpu",
|
|
"dev:rocm": "bash dev.sh --rocm",
|
|
"dev:server": "bash server/start.sh",
|
|
"dev:server:cpu": "bash server/start.sh --cpu",
|
|
"dev:server:rocm": "bash server/start.sh --rocm",
|
|
"dev:web": "pnpm --filter vibepod-web dev",
|
|
"format": "prettier --write . && cd server && uv run ruff format .",
|
|
"format:check": "prettier --check . && cd server && uv run ruff format --check .",
|
|
"lint:server": "cd server && uv run ruff check .",
|
|
"lint:server:fix": "cd server && uv run ruff check --fix ."
|
|
},
|
|
"devDependencies": {
|
|
"prettier": "^3.5.3"
|
|
},
|
|
"packageManager": "pnpm@10.33.2+sha512.a90faf6feeab71ad6c6e57f94e0fe1a12f5dcc22cd754db40ae9593eb6a3e0b6b12e3540218bb37ae083404b1f2ce6db2a4121e979829b4aff94b99f49da1cf8"
|
|
}
|