blackteay/GPT-SoVITS
1
0
Fork 0
mirror of https://github.com/RVC-Boss/GPT-SoVITS.git synced 2026-09-15 05:35:42 +08:00
Code Issues Packages Projects Releases Wiki Activity

Merge 1b8059cf610c1e89bd7798cab721145ea1fc50a6 into 48b1a0169a28582a8984402f82cf438d3bfa6aca

Browse Source
This commit is contained in:
li-lizhe 2026-09-04 10:56:31 +08:00 committed by GitHub
commit 1e150aef3c
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 9 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
GPT_SoVITS/export_torch_script_v3v4.py
View File

@ -27,7 +27,15 @@ logging.config.dictConfig(uvicorn.config.LOGGING_CONFIG)
logger = logging.getLogger("uvicorn")
is_half = True
device = "cuda" if torch.cuda.is_available() else "cpu"
# Detect the current accelerator (CUDA, NPU, XPU, MPS, …) in a device-agnostic way.
# Using torch.accelerator (available since PyTorch 2.5) ensures the script works
# out of the box on Ascend NPU, Intel XPU, Apple MPS, etc. without hard-coding
# "cuda". Falls back to the traditional CUDA-or-CPU heuristic when torch.accelerator
# is not available (PyTorch < 2.5).
if hasattr(torch, "accelerator") and torch.accelerator.is_available():
device = str(torch.accelerator.current_accelerator())
else:
device = "cuda" if torch.cuda.is_available() else "cpu"
now_dir = os.getcwd()