blackteay/GPT-SoVITS
1
0
Fork 0
mirror of https://github.com/RVC-Boss/GPT-SoVITS.git synced 2025-12-16 17:40:09 +08:00
Code Issues Packages Projects Releases Wiki Activity

solved subtle problem on import and syntax

Browse Source
This commit is contained in:
plae-tljg 2025-10-20 20:28:49 +08:00
parent 06994f2a13
commit 42abb66a32
3 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
GPT_SoVITS/s2_train.py
View File

@ -17,7 +17,7 @@ import torch.multiprocessing as mp
from torch.cuda.amp import GradScaler from torch.cuda.amp import GradScaler
musa_ddp = False musa_ddp = False
if musa_utils.is_available(): if musa_utils.is_available():
os.environ["MUSA_LAUNCH_BLOCKING"] = 1 os.environ["MUSA_LAUNCH_BLOCKING"] = "1"
autocast = torch.musa.amp.autocast autocast = torch.musa.amp.autocast
musa_ddp = musa_utils.should_ddp() musa_ddp = musa_utils.should_ddp()
elif torch.cuda.is_available(): elif torch.cuda.is_available():

2
musa_accelerator.py
View File

@ -18,6 +18,8 @@ from lightning_fabric.utilities.types import _DEVICE
from pytorch_lightning.accelerators.accelerator import Accelerator from pytorch_lightning.accelerators.accelerator import Accelerator
from pytorch_lightning.utilities.exceptions import MisconfigurationException from pytorch_lightning.utilities.exceptions import MisconfigurationException
from musa_utils import _musa_available
import logging import logging
_log = logging.getLogger(__name__) _log = logging.getLogger(__name__)

3
musa_utils.py
View File

@ -68,8 +68,7 @@ def get_device_dtype(device_idx: int = 0) -> tuple[torch.dtype, float, float]:
return dtype, version_val, mem_gb return dtype, version_val, mem_gb
def should_ddp(device_idx: int = 0) -> bool: def should_ddp(device_idx: int = 0) -> bool:
device_name = torch.musa.get_device_properties(device_idx).name _, version_val, _ = get_device_dtype(device_idx)
numbers_in_name = [int(n) for n in re.findall(r'\d+', device_name)]
if version_val <= 2.1: if version_val <= 2.1:
return False return False
else: else: